reab5555 commited on
Commit
baf1719
·
verified ·
1 Parent(s): 4bc1a25

Update processing.py

Browse files
Files changed (1) hide show
  1. processing.py +4 -4
processing.py CHANGED
@@ -47,10 +47,10 @@ class LazyPipeline:
47
  "text-generation",
48
  model=model,
49
  tokenizer=tokenizer,
50
- max_new_tokens=1000,
51
- temperature=0.1,
52
- top_p = 0.99,
53
- top_k = 1,
54
  repetition_penalty = 1.2,
55
  )
56
  return self.pipeline
 
47
  "text-generation",
48
  model=model,
49
  tokenizer=tokenizer,
50
+ max_new_tokens=800,
51
+ temperature=0.3,
52
+ top_p = 0.95,
53
+ top_k = 5,
54
  repetition_penalty = 1.2,
55
  )
56
  return self.pipeline