pingnie commited on
Commit
5f34a5c
1 Parent(s): 17162c6

add debug info

Browse files
Files changed (1) hide show
  1. src/backend/manage_requests.py +1 -1
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