Menelthornil commited on
Commit
e689201
·
verified ·
1 Parent(s): d787ad6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -18,6 +18,7 @@ model = AutoGPTQForCausalLM.from_quantized(
18
  trust_remote_code=True,
19
  device_map="auto", # Use auto to ensure compatibility with CPU
20
  use_safetensors=True,
 
21
  quantize_config=None,
22
  )
23
 
 
18
  trust_remote_code=True,
19
  device_map="auto", # Use auto to ensure compatibility with CPU
20
  use_safetensors=True,
21
+ torch_dtype=torch.float32, # Use float32 for CPU compatibility
22
  quantize_config=None,
23
  )
24