nm-testing
/

llama2-7b-gsm8k-pt-pruned50-quant-ds

Text Generation

text-generation-inference

Model card Files Files and versions Community

mwitiderrick commited on Feb 2

Commit

a98c146

•

1 Parent(s): 6f7231f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -26,7 +26,7 @@ from deepsparse import TextGeneration
 prompt = "James decides to run 3 sprints 3 times a week. He runs 60 meters each sprint. How many total meters does he run a week?"
 formatted_prompt =  f"Question:{prompt}\nAnswer:"
-model = TextGeneration(model_path="hf:nm-testing/TinyLlama-1.1B-intermediate-step-1431k-3T-gsms8k-pruned50-quant-ds")
 print(model(formatted_prompt, max_new_tokens=200).generations[0].text)
 """
 First find the total distance of one sprint: 60 meters * 3 = <<60*3=180>>180 meters

 prompt = "James decides to run 3 sprints 3 times a week. He runs 60 meters each sprint. How many total meters does he run a week?"
 formatted_prompt =  f"Question:{prompt}\nAnswer:"
+model = TextGeneration(model_path="hf:nm-testing/llama2-7b-gsm8k-pt-pruned50-quant-ds")
 print(model(formatted_prompt, max_new_tokens=200).generations[0].text)
 """
 First find the total distance of one sprint: 60 meters * 3 = <<60*3=180>>180 meters