mwitiderrick commited on
Commit
6350b85
·
1 Parent(s): 01a8101

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -29,7 +29,7 @@ user_message = 'How many days are there in a leap year?'
29
 
30
  formatted_prompt = f'### System:\n{system_message}<|endoftext|>\n### User:\n{user_message}<|endoftext|>\n### Assistant:\n'
31
 
32
- model = TextGeneration(model_path="hf:nm-testing/open_llama_3b-pruned-quant-50")
33
  print(model(formatted_prompt, max_new_tokens=200,repetition_penalty=1.05).generations[0].text)
34
 
35
  """
 
29
 
30
  formatted_prompt = f'### System:\n{system_message}<|endoftext|>\n### User:\n{user_message}<|endoftext|>\n### Assistant:\n'
31
 
32
+ model = TextGeneration(model_path="hf:nm-testing/open-llama-3b-v2-instruct-pruned50-quant-ds")
33
  print(model(formatted_prompt, max_new_tokens=200,repetition_penalty=1.05).generations[0].text)
34
 
35
  """