Commit
·
4069984
1
Parent(s):
5ee8db6
Update minigpt4/models/mini_gpt4.py
Browse files
minigpt4/models/mini_gpt4.py
CHANGED
@@ -212,7 +212,7 @@ class MiniGPT4(Blip2Base):
|
|
212 |
@classmethod
|
213 |
def from_config(cls, cfg):
|
214 |
vit_model = cfg.get("vit_model", "eva_clip_g")
|
215 |
-
q_former_model = cfg.get("q_former_model", "
|
216 |
img_size = cfg.get("image_size")
|
217 |
num_query_token = cfg.get("num_query_token")
|
218 |
llama_model = cfg.get("llama_model")
|
|
|
212 |
@classmethod
|
213 |
def from_config(cls, cfg):
|
214 |
vit_model = cfg.get("vit_model", "eva_clip_g")
|
215 |
+
q_former_model = cfg.get("q_former_model", "blip2_pretrained_flant5xxl.pth")
|
216 |
img_size = cfg.get("image_size")
|
217 |
num_query_token = cfg.get("num_query_token")
|
218 |
llama_model = cfg.get("llama_model")
|