X-iZhang commited on
Commit
3e1e6a0
·
verified ·
1 Parent(s): 034d147

Update libra/model/builder.py

Browse files
Files changed (1) hide show
  1. libra/model/builder.py +1 -1
libra/model/builder.py CHANGED
@@ -23,7 +23,7 @@ from libra.model import *
23
  from libra.constants import DEFAULT_IMAGE_PATCH_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN
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
 
 
23
  from libra.constants import DEFAULT_IMAGE_PATCH_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN
24
 
25
 
26
+ def load_pretrained_model(model_path, model_base, model_name, load_8bit=False, load_4bit=False, device_map="auto", device="cpu"):
27
  kwargs = {"device_map": device_map}
28
 
29