pratikroy311 commited on
Commit
49fa231
·
verified ·
1 Parent(s): c6e8170

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +0 -1
utils.py CHANGED
@@ -57,7 +57,6 @@ text_generation_pipeline = pipeline(
57
  repetition_penalty=1.1,
58
  return_full_text=True,
59
  max_new_tokens=400,
60
- inference= True,
61
  )
62
 
63
  llm = HuggingFacePipeline(pipeline=text_generation_pipeline)
 
57
  repetition_penalty=1.1,
58
  return_full_text=True,
59
  max_new_tokens=400,
 
60
  )
61
 
62
  llm = HuggingFacePipeline(pipeline=text_generation_pipeline)