anasmkh commited on
Commit
2715f42
·
1 Parent(s): a697213

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -29,7 +29,7 @@ pipeline = pipeline(
29
  "text-generation",
30
  model=model,
31
  tokenizer=tokenizer,
32
- # max_length=64
33
  )
34
 
35
  # local_llm = HuggingFacePipeline(pipeline=pipeline)
 
29
  "text-generation",
30
  model=model,
31
  tokenizer=tokenizer,
32
+ max_length=512
33
  )
34
 
35
  # local_llm = HuggingFacePipeline(pipeline=pipeline)