neuralmagic
/

Llama-2-7b-ultrachat200k-pruned_70-quantized-deepsparse

Text Generation

text-generation-inference

Model card Files Files and versions Community

mgoin commited on Mar 18

Commit

4cea4f3

•

1 Parent(s): 52efa4e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -38,7 +38,7 @@ from deepsparse import TextGeneration
 model = TextGeneration(model_path="hf:neuralmagic/Llama-2-7b-pruned70-retrained-ultrachat-quant-ds")
 input_text = "Write me a poem about Machine Learning."
-outputs = model(formatted_prompt, max_new_tokens=100)
 print(outputs.generations[0].text)
 ```

 model = TextGeneration(model_path="hf:neuralmagic/Llama-2-7b-pruned70-retrained-ultrachat-quant-ds")
 input_text = "Write me a poem about Machine Learning."
+outputs = model(input_text, max_new_tokens=100)
 print(outputs.generations[0].text)
 ```