Spaces:
Runtime error
Runtime error
Commit
Β·
b104280
1
Parent(s):
6feb8b1
Update minigpt4/models/mini_gpt4.py
Browse files
minigpt4/models/mini_gpt4.py
CHANGED
|
@@ -30,7 +30,7 @@ class MiniGPT4(Blip2Base):
|
|
| 30 |
def __init__(
|
| 31 |
self,
|
| 32 |
vit_model="eva_clip_g",
|
| 33 |
-
q_former_model="blip2_pretrained_flant5xxl.pth",
|
| 34 |
img_size=224,
|
| 35 |
drop_path_rate=0,
|
| 36 |
use_grad_checkpoint=False,
|
|
@@ -221,7 +221,7 @@ class MiniGPT4(Blip2Base):
|
|
| 221 |
@classmethod
|
| 222 |
def from_config(cls, cfg):
|
| 223 |
vit_model = cfg.get("vit_model", "eva_clip_g")
|
| 224 |
-
q_former_model = cfg.get("q_former_model", "blip2_pretrained_flant5xxl.pth")
|
| 225 |
img_size = cfg.get("image_size")
|
| 226 |
num_query_token = cfg.get("num_query_token")
|
| 227 |
llama_model = cfg.get("llama_model")
|
|
|
|
| 30 |
def __init__(
|
| 31 |
self,
|
| 32 |
vit_model="eva_clip_g",
|
| 33 |
+
q_former_model="https://storage.googleapis.com/sfr-vision-language-research/LAVIS/models/BLIP2/blip2_pretrained_flant5xxl.pth",
|
| 34 |
img_size=224,
|
| 35 |
drop_path_rate=0,
|
| 36 |
use_grad_checkpoint=False,
|
|
|
|
| 221 |
@classmethod
|
| 222 |
def from_config(cls, cfg):
|
| 223 |
vit_model = cfg.get("vit_model", "eva_clip_g")
|
| 224 |
+
q_former_model = cfg.get("q_former_model", "https://storage.googleapis.com/sfr-vision-language-research/LAVIS/models/BLIP2/blip2_pretrained_flant5xxl.pth")
|
| 225 |
img_size = cfg.get("image_size")
|
| 226 |
num_query_token = cfg.get("num_query_token")
|
| 227 |
llama_model = cfg.get("llama_model")
|