lkjjj26 commited on
Commit
23e94ce
·
verified ·
1 Parent(s): 7e83b48

update max token 1024

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ class PDBSearchAssistant:
30
  self.pipe = pipeline(
31
  "text2text-generation",
32
  model=model_name,
33
- max_new_tokens=512,
34
  temperature=0.3,
35
  torch_dtype="auto",
36
  device="cpu"
 
30
  self.pipe = pipeline(
31
  "text2text-generation",
32
  model=model_name,
33
+ max_new_tokens=1024,
34
  temperature=0.3,
35
  torch_dtype="auto",
36
  device="cpu"