aksinghyaani commited on
Commit
655e347
·
verified ·
1 Parent(s): 94122b8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -118,7 +118,7 @@ model_inputs = tokenizer([prompt], return_tensors="pt").to(model.device)
118
 
119
  outputs = model.generate(
120
  **model_inputs,
121
- max_new_tokens=200,
122
  do_sample=True,
123
  temperature=0.3,
124
  top_k=20,
 
118
 
119
  outputs = model.generate(
120
  **model_inputs,
121
+ max_new_tokens=50,
122
  do_sample=True,
123
  temperature=0.3,
124
  top_k=20,