Spaces:
Running
Running
Update libra/model/builder.py
Browse files- 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=
|
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 |
|