Spaces:
Running
Running
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,
|