alaselababatunde commited on
Commit
02f04dd
·
1 Parent(s): 00ccd59
Files changed (1) hide show
  1. main.py +5 -5
main.py CHANGED
@@ -45,11 +45,11 @@ try:
45
  "text2text-generation",
46
  model=MODEL_NAME,
47
  device=DEVICE,
48
- max_new_tokens=1024, # give enough room for thoughtful answers
49
- temperature=0.7, # higher temp = more natural and creative
50
- do_sample=True, # sample to allow variety
51
- top_p=0.9, # nucleus sampling to filter unlikely tokens
52
- repetition_penalty=1.2 # avoid repeating the same phrases
53
  )
54
 
55
 
 
45
  "text2text-generation",
46
  model=MODEL_NAME,
47
  device=DEVICE,
48
+ max_new_tokens=1024,
49
+ temperature=0.7,
50
+ do_sample=True,
51
+ top_p=0.9,
52
+ repetition_penalty=1.2
53
  )
54
 
55