mwitiderrick
commited on
Commit
•
a0100d5
1
Parent(s):
b55e5b3
Update README.md
Browse files
README.md
CHANGED
@@ -25,9 +25,9 @@ Run in a [Python pipeline](https://github.com/neuralmagic/deepsparse/blob/main/d
|
|
25 |
from deepsparse import TextGeneration
|
26 |
|
27 |
prompt = "How to make banana bread?"
|
28 |
-
formatted_prompt = f"
|
29 |
|
30 |
-
model = TextGeneration(model_path="hf:nm-testing/
|
31 |
|
32 |
print(model(formatted_prompt, max_new_tokens=200).generations[0].text)
|
33 |
"""
|
|
|
25 |
from deepsparse import TextGeneration
|
26 |
|
27 |
prompt = "How to make banana bread?"
|
28 |
+
formatted_prompt = f"<s>[INST]{prompt}[/INST]"
|
29 |
|
30 |
+
model = TextGeneration(model_path="hf:nm-testing/Llama-2-7b-hf-pruned50-quant-ds")
|
31 |
|
32 |
print(model(formatted_prompt, max_new_tokens=200).generations[0].text)
|
33 |
"""
|