Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -150,7 +150,7 @@ def generate_deepseek(prompt: str) -> (str, str):
|
|
| 150 |
inputs = deepseek_tokenizer(prompt, return_tensors="pt", truncation=True, max_length=1024).to(device)
|
| 151 |
outputs = deepseek_model.generate(
|
| 152 |
**inputs,
|
| 153 |
-
|
| 154 |
temperature=0.7,
|
| 155 |
top_p=0.9,
|
| 156 |
do_sample=True,
|
|
|
|
| 150 |
inputs = deepseek_tokenizer(prompt, return_tensors="pt", truncation=True, max_length=1024).to(device)
|
| 151 |
outputs = deepseek_model.generate(
|
| 152 |
**inputs,
|
| 153 |
+
max_new_tokens=512,
|
| 154 |
temperature=0.7,
|
| 155 |
top_p=0.9,
|
| 156 |
do_sample=True,
|