arjunanand13 commited on
Commit
4777fa9
1 Parent(s): 2477cb2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -49,10 +49,10 @@ generate_text = pipeline(
49
  tokenizer=tokenizer,
50
  return_full_text=True,
51
  task='text-generation',
52
- stopping_criteria=stopping_criteria,
53
  temperature=0.1,
54
- max_new_tokens=512,
55
- repetition_penalty=1.1
56
  )
57
 
58
  llm = HuggingFacePipeline(pipeline=generate_text)
 
49
  tokenizer=tokenizer,
50
  return_full_text=True,
51
  task='text-generation',
52
+ # stopping_criteria=stopping_criteria,
53
  temperature=0.1,
54
+ max_new_tokens=2048,
55
+ # repetition_penalty=1.1
56
  )
57
 
58
  llm = HuggingFacePipeline(pipeline=generate_text)