Spaces:
Runtime error
Runtime error
pminervini
commited on
Commit
•
8e10a29
1
Parent(s):
60529bb
update
Browse files
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}"
|