Ensuring safe responses
You're configuring an internal chatbot for a medical team. To ensure consistent responses, you need to limit variability by setting a token limit and restricting token selection.
You have been provided the Llama class instance in the llm variable and the code to call the completion. You are also given a sample prompt to test with.
This exercise is part of the course
Working with Llama 3
Exercise instructions
- Set the model parameters so that the maximum number of tokens is limited to ten tokens, and the model only ever chooses between the two most likely words at each completion step.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
output = llm(
"What are the symptoms of strep throat?",
# Set the model parameters
max_tokens=____, #Limit response length
top_k=____ #Restrict word choices
)
print(output['choices'][0]['text'])