Spaces:
Sleeping
Sleeping
saicharan1234
commited on
Update minigpt4/models/mini_gpt4.py
Browse files
minigpt4/models/mini_gpt4.py
CHANGED
@@ -219,7 +219,7 @@ class MiniGPT4(Blip2Base):
|
|
219 |
@classmethod
|
220 |
def from_config(cls, cfg):
|
221 |
vit_model = cfg.get("vit_model", "eva_clip_g")
|
222 |
-
q_former_model = cfg.get("q_former_model", "
|
223 |
img_size = cfg.get("image_size")
|
224 |
num_query_token = cfg.get("num_query_token")
|
225 |
llama_model = cfg.get("llama_model")
|
|
|
219 |
@classmethod
|
220 |
def from_config(cls, cfg):
|
221 |
vit_model = cfg.get("vit_model", "eva_clip_g")
|
222 |
+
q_former_model = cfg.get("q_former_model", "blip2_pretrained_flant5xxl.pth")
|
223 |
img_size = cfg.get("image_size")
|
224 |
num_query_token = cfg.get("num_query_token")
|
225 |
llama_model = cfg.get("llama_model")
|