gouravbhadraDev commited on
Commit
ad10382
·
verified ·
1 Parent(s): 5afcefa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- max_length=1024,
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,