NickyNicky commited on
Commit
c4687d7
1 Parent(s): 11a43b2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -168,7 +168,9 @@ What is the meaning of life in the current time?<end_of_turn>
168
  <start_of_turn>model
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,
 
168
  <start_of_turn>model
169
  """
170
 
171
+ inputs = tokenizer.encode(txt,
172
+ return_tensors="pt",
173
+ add_special_tokens=False).to("cuda:0")
174
  max_new_tokens=1000
175
  generation_config = GenerationConfig(
176
  max_new_tokens=max_new_tokens,