Hugging Face's logo

Spaces:
X-iZhang
/
Libra
Running

App Files Community
1
X-iZhang commited on
Commit
4de70fa
·
verified ·
1 Parent(s): 43bcde1

Update libra/model/builder.py

Browse files
Files changed (1) hide show
  1. libra/model/builder.py +1 -1
libra/model/builder.py CHANGED
@@ -24,7 +24,7 @@ from libra.constants import DEFAULT_IMAGE_PATCH_TOKEN, DEFAULT_IM_START_TOKEN, D
24
 
25
 
26
  def load_pretrained_model(model_path, model_base, model_name, load_8bit=False, load_4bit=True, device_map="auto", device="cpu"):
27
- kwargs = {"device_map": device_map, **kwargs}
28
 
29
 
30
  if device != "cuda":
 
24
 
25
 
26
  def load_pretrained_model(model_path, model_base, model_name, load_8bit=False, load_4bit=True, device_map="auto", device="cpu"):
27
+ kwargs = {"device_map": device_map}
28
 
29
 
30
  if device != "cuda":