jdesiree commited on
Commit
331783c
·
verified ·
1 Parent(s): 0a58c6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ llm = HuggingFaceEndpoint(
22
  temperature=0.7,
23
  top_p=0.9,
24
  repetition_penalty=1.1,
25
- model_kwargs={"max_length": 1024},
26
  huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
27
  )
28
 
 
22
  temperature=0.7,
23
  top_p=0.9,
24
  repetition_penalty=1.1,
25
+ max_new_tokens=1024,
26
  huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
27
  )
28