Vasily Alexeev commited on
Commit
0606c7d
·
1 Parent(s): 89f2cc0

remove commented piece from code example

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -198,7 +198,7 @@ outputs = model.generate(
198
  do_sample=True, temperature=0.7, top_p=0.95,
199
  )
200
 
201
- response = tokenizer.decode(outputs[0]) # , skip_special_tokens=True)
202
  continuation = response.removeprefix(prompt).removesuffix(tokenizer.eos_token)
203
 
204
  print(f'Prompt:\n{prompt}')
 
198
  do_sample=True, temperature=0.7, top_p=0.95,
199
  )
200
 
201
+ response = tokenizer.decode(outputs[0])
202
  continuation = response.removeprefix(prompt).removesuffix(tokenizer.eos_token)
203
 
204
  print(f'Prompt:\n{prompt}')