Spaces:
Running
Running
add debug info
Browse files
src/backend/manage_requests.py
CHANGED
|
@@ -31,7 +31,7 @@ class EvalRequest:
|
|
| 31 |
|
| 32 |
def get_model_args(self) -> str:
|
| 33 |
model_args = f"pretrained={self.model},revision={self.revision},parallelize=True" # ,max_length=4096"
|
| 34 |
-
model_args += ",trust_remote_code=True"
|
| 35 |
if self.precision in ["float16", "float32", "bfloat16"]:
|
| 36 |
model_args += f",dtype={self.precision}"
|
| 37 |
# Quantized models need some added config, the install of bits and bytes, etc
|
|
|
|
| 31 |
|
| 32 |
def get_model_args(self) -> str:
|
| 33 |
model_args = f"pretrained={self.model},revision={self.revision},parallelize=True" # ,max_length=4096"
|
| 34 |
+
model_args += ",trust_remote_code=True,device_map=auto"
|
| 35 |
if self.precision in ["float16", "float32", "bfloat16"]:
|
| 36 |
model_args += f",dtype={self.precision}"
|
| 37 |
# Quantized models need some added config, the install of bits and bytes, etc
|