pminervini commited on
Commit
8e10a29
1 Parent(s): 60529bb
Files changed (1) hide show
  1. src/backend/manage_requests.py +1 -1
src/backend/manage_requests.py CHANGED
@@ -26,7 +26,7 @@ class EvalRequest:
26
  license: Optional[str] = ""
27
 
28
  def get_model_args(self) -> str:
29
- model_args = f"pretrained={self.model},revision={self.revision},parallelize=True"
30
 
31
  if self.precision in ["float16", "float32", "bfloat16"]:
32
  model_args += f",dtype={self.precision}"
 
26
  license: Optional[str] = ""
27
 
28
  def get_model_args(self) -> str:
29
+ model_args = f"pretrained={self.model},revision={self.revision},parallelize=True,max_length=4096"
30
 
31
  if self.precision in ["float16", "float32", "bfloat16"]:
32
  model_args += f",dtype={self.precision}"