update max token 1024
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ class PDBSearchAssistant:
|
|
30 |
self.pipe = pipeline(
|
31 |
"text2text-generation",
|
32 |
model=model_name,
|
33 |
-
max_new_tokens=
|
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"
|