Matej commited on
Commit
3617df8
·
verified ·
1 Parent(s): a1a9fc7

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -53,7 +53,7 @@ inputs = processor.apply_chat_template(
53
  input_len = inputs["input_ids"].shape[-1]
54
 
55
  with torch.inference_mode():
56
- generation = model.generate(**inputs, max_new_tokens=500, do_sample=False)
57
  generation = generation[0][input_len:]
58
 
59
  decoded = processor.decode(generation, skip_special_tokens=True)
 
53
  input_len = inputs["input_ids"].shape[-1]
54
 
55
  with torch.inference_mode():
56
+ generation = model.generate(**inputs, max_new_tokens=500)
57
  generation = generation[0][input_len:]
58
 
59
  decoded = processor.decode(generation, skip_special_tokens=True)