mwitiderrick commited on
Commit
263e235
1 Parent(s): db25206

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -27,7 +27,7 @@ from deepsparse import TextGeneration
27
  prompt = "How to make banana bread?"
28
  formatted_prompt = f"<|im_start|>user\n{prompt}<|im_end|>\n<|im_start|>assistant\n"
29
 
30
- model = TextGeneration(model="hf:neuralmagic/TinyLlama-1.1B-Chat-v0.4-pruned50-quant-ds")
31
  print(model(formatted_prompt, max_new_tokens=500).generations[0].text)
32
 
33
  """
 
27
  prompt = "How to make banana bread?"
28
  formatted_prompt = f"<|im_start|>user\n{prompt}<|im_end|>\n<|im_start|>assistant\n"
29
 
30
+ model = TextGeneration(model="hf:nm-testing/TinyLlama-1.1B-Chat-v1.0-pruned50-quant-ds")
31
  print(model(formatted_prompt, max_new_tokens=500).generations[0].text)
32
 
33
  """