Upload folder using huggingface_hub
Browse files
server.py
CHANGED
|
@@ -77,7 +77,7 @@ def run_inference(text: str) -> str:
|
|
| 77 |
with torch.no_grad():
|
| 78 |
outputs = model.generate(
|
| 79 |
**inputs,
|
| 80 |
-
max_new_tokens=
|
| 81 |
temperature=0.7,
|
| 82 |
do_sample=True,
|
| 83 |
top_p=0.9,
|
|
|
|
| 77 |
with torch.no_grad():
|
| 78 |
outputs = model.generate(
|
| 79 |
**inputs,
|
| 80 |
+
max_new_tokens=256,
|
| 81 |
temperature=0.7,
|
| 82 |
do_sample=True,
|
| 83 |
top_p=0.9,
|