Commit
·
6350b85
1
Parent(s):
01a8101
Update README.md
Browse files
README.md
CHANGED
@@ -29,7 +29,7 @@ user_message = 'How many days are there in a leap year?'
|
|
29 |
|
30 |
formatted_prompt = f'### System:\n{system_message}<|endoftext|>\n### User:\n{user_message}<|endoftext|>\n### Assistant:\n'
|
31 |
|
32 |
-
model = TextGeneration(model_path="hf:nm-testing/
|
33 |
print(model(formatted_prompt, max_new_tokens=200,repetition_penalty=1.05).generations[0].text)
|
34 |
|
35 |
"""
|
|
|
29 |
|
30 |
formatted_prompt = f'### System:\n{system_message}<|endoftext|>\n### User:\n{user_message}<|endoftext|>\n### Assistant:\n'
|
31 |
|
32 |
+
model = TextGeneration(model_path="hf:nm-testing/open-llama-3b-v2-instruct-pruned50-quant-ds")
|
33 |
print(model(formatted_prompt, max_new_tokens=200,repetition_penalty=1.05).generations[0].text)
|
34 |
|
35 |
"""
|