Update README.md
Browse files
README.md
CHANGED
@@ -27,7 +27,7 @@ from deepsparse import TextGeneration
|
|
27 |
prompt = 'Q: What is the largest animal?\nA:'
|
28 |
formatted_prompt = f"Q: {prompt}\nA:"
|
29 |
|
30 |
-
model = TextGeneration(model_path="hf:nm-testing/open_llama_3b-
|
31 |
print(model(formatted_prompt, max_new_tokens=200).generations[0].text)
|
32 |
|
33 |
"""
|
|
|
27 |
prompt = 'Q: What is the largest animal?\nA:'
|
28 |
formatted_prompt = f"Q: {prompt}\nA:"
|
29 |
|
30 |
+
model = TextGeneration(model_path="hf:nm-testing/open_llama_3b-pruned50-quant-ds")
|
31 |
print(model(formatted_prompt, max_new_tokens=200).generations[0].text)
|
32 |
|
33 |
"""
|