Commit
·
507bf37
1
Parent(s):
2ee5da2
Update README.md
Browse files
README.md
CHANGED
@@ -10,7 +10,7 @@ from deepsparse import TextGeneration
|
|
10 |
prompt = "How to get in a good university?"
|
11 |
formatted_prompt = f"<s> [|User|]\n{prompt}</s>[|Assistant|]\n"
|
12 |
|
13 |
-
model = TextGeneration(model="
|
14 |
|
15 |
print(model(formatted_prompt, max_new_tokens=500).generations[0].text)
|
16 |
"""
|
@@ -29,7 +29,7 @@ from deepsparse import TextGeneration
|
|
29 |
prompt = "How to become a great software engineer?"
|
30 |
formatted_prompt = f"<s> [|User|]\n{prompt}</s>[|Assistant|]\n"
|
31 |
|
32 |
-
model = TextGeneration(model="
|
33 |
|
34 |
print(model(formatted_prompt, max_new_tokens=500).generations[0].text)
|
35 |
"""
|
|
|
10 |
prompt = "How to get in a good university?"
|
11 |
formatted_prompt = f"<s> [|User|]\n{prompt}</s>[|Assistant|]\n"
|
12 |
|
13 |
+
model = TextGeneration(model="hf:nm-testing/MiniChat-3B-pruned-quant-50")
|
14 |
|
15 |
print(model(formatted_prompt, max_new_tokens=500).generations[0].text)
|
16 |
"""
|
|
|
29 |
prompt = "How to become a great software engineer?"
|
30 |
formatted_prompt = f"<s> [|User|]\n{prompt}</s>[|Assistant|]\n"
|
31 |
|
32 |
+
model = TextGeneration(model="hf:nm-testing/MiniChat-3B-pruned-quant-50")
|
33 |
|
34 |
print(model(formatted_prompt, max_new_tokens=500).generations[0].text)
|
35 |
"""
|