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
|