NickyNicky commited on
Commit
db8d789
1 Parent(s): 8755dc4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -169,7 +169,7 @@ What is the meaning of life in the current time?<end_of_turn>
169
  """
170
 
171
  inputs = tokenizer.encode(txt, return_tensors="pt").to("cuda")
172
-
173
  generation_config = GenerationConfig(
174
  max_new_tokens=max_new_tokens,
175
  temperature=0.55,
 
169
  """
170
 
171
  inputs = tokenizer.encode(txt, return_tensors="pt").to("cuda")
172
+ max_new_tokens=1000
173
  generation_config = GenerationConfig(
174
  max_new_tokens=max_new_tokens,
175
  temperature=0.55,