Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -153,7 +153,7 @@ class CreedBrattonAI:
|
|
| 153 |
with torch.no_grad():
|
| 154 |
outputs = self.model.generate(
|
| 155 |
inputs,
|
| 156 |
-
max_new_tokens=
|
| 157 |
do_sample=True,
|
| 158 |
temperature=0.9,
|
| 159 |
top_p=0.95,
|
|
|
|
| 153 |
with torch.no_grad():
|
| 154 |
outputs = self.model.generate(
|
| 155 |
inputs,
|
| 156 |
+
max_new_tokens=4000,
|
| 157 |
do_sample=True,
|
| 158 |
temperature=0.9,
|
| 159 |
top_p=0.95,
|