reab5555 commited on
Commit
5c84887
·
verified ·
1 Parent(s): 30efc24

Update processing.py

Browse files
Files changed (1) hide show
  1. processing.py +5 -1
processing.py CHANGED
@@ -43,7 +43,11 @@ class LazyPipeline:
43
  model=model,
44
  tokenizer=tokenizer,
45
  max_new_tokens=800,
46
- temperature=0.8,
 
 
 
 
47
  )
48
  return self.pipeline
49
 
 
43
  model=model,
44
  tokenizer=tokenizer,
45
  max_new_tokens=800,
46
+ temperature=0.3,
47
+ top_p = 0.95,
48
+ top_k = 5,
49
+ repetition_penalty = 1.2,
50
+ do_sample=True,
51
  )
52
  return self.pipeline
53