neuralmagic
/

Llama-2-7b-dolphin-open_platypus-pruned_50-quantized-deepsparse

Text Generation

Model card Files Files and versions Community

mgoin commited on Mar 18

Commit

1fd6b52

•

1 Parent(s): 2929313

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -39,7 +39,7 @@ from deepsparse import TextGeneration
 model = TextGeneration(model_path="hf:neuralmagic/Llama-2-7b-pruned50-retrained-instruct-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-pruned50-retrained-instruct-quant-ds")
 input_text = "Write me a poem about Machine Learning."
+outputs = model(input_text, max_new_tokens=100)
 print(outputs.generations[0].text)
 ```