Soh19 commited on
Commit
f8b93ba
·
verified ·
1 Parent(s): 16f91ef

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -112,7 +112,7 @@ text = """
112
 
113
  # Tokenize input and generate output
114
  inputs = tokenizer(text, return_tensors="pt")
115
- outputs = model.generate(**inputs.to(device), max_new_tokens=185)
116
 
117
  # Decode the output
118
  result = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
112
 
113
  # Tokenize input and generate output
114
  inputs = tokenizer(text, return_tensors="pt")
115
+ outputs = model.generate(**inputs.to("cuda"), max_new_tokens=185,do_sample=False,top_k=None,temperature=1.0,top_p=None)
116
 
117
  # Decode the output
118
  result = tokenizer.decode(outputs[0], skip_special_tokens=True)