marialasam commited on
Commit
12de5b8
·
verified ·
1 Parent(s): df5c4aa

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -64,7 +64,7 @@ text = "When faced with a complex problem, what should one do?"
64
  input_ids = tokenizer(text, return_tensors="pt").to('cuda').input_ids
65
  outputs = model.generate(
66
  input_ids,
67
- max_length=1000,
68
  temperature=0.6,
69
  top_p=0.9,
70
  repetition_penalty=1.2,
 
64
  input_ids = tokenizer(text, return_tensors="pt").to('cuda').input_ids
65
  outputs = model.generate(
66
  input_ids,
67
+ max_length=2500,
68
  temperature=0.6,
69
  top_p=0.9,
70
  repetition_penalty=1.2,