Commit
·
868d4cc
1
Parent(s):
c08ab20
Update README.md
Browse files
README.md
CHANGED
|
@@ -41,7 +41,7 @@ from deepsparse import TextGeneration
|
|
| 41 |
prompt = "How to become a great software engineer?"
|
| 42 |
formatted_prompt = f"<s> [|User|]\n{prompt}</s>[|Assistant|]\n"
|
| 43 |
|
| 44 |
-
model = TextGeneration(model="hf:nm-testing/MiniChat-3B-
|
| 45 |
|
| 46 |
print(model(formatted_prompt, max_new_tokens=500).generations[0].text)
|
| 47 |
"""
|
|
|
|
| 41 |
prompt = "How to become a great software engineer?"
|
| 42 |
formatted_prompt = f"<s> [|User|]\n{prompt}</s>[|Assistant|]\n"
|
| 43 |
|
| 44 |
+
model = TextGeneration(model="hf:nm-testing/MiniChat-3B-pruned50-quant-ds")
|
| 45 |
|
| 46 |
print(model(formatted_prompt, max_new_tokens=500).generations[0].text)
|
| 47 |
"""
|