Soh19 commited on
Commit
8192e91
·
verified ·
1 Parent(s): 38b315f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -106,7 +106,7 @@ text = """
106
 
107
  # Tokenize input and generate output
108
  inputs = tokenizer(text, return_tensors="pt")
109
- outputs = model.generate(**inputs.to(device), max_new_tokens=200)
110
 
111
  # Decode the output
112
  result = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
106
 
107
  # Tokenize input and generate output
108
  inputs = tokenizer(text, return_tensors="pt")
109
+ outputs = model.generate(**inputs.to(device), max_new_tokens=185)
110
 
111
  # Decode the output
112
  result = tokenizer.decode(outputs[0], skip_special_tokens=True)