Spaces:
Runtime error
Runtime error
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")
|