nmysore commited on
Commit
5b2616c
·
verified ·
1 Parent(s): b692662

Upload folder using huggingface_hub

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