Spaces:
Runtime error
Runtime error
Vision-CAIR
commited on
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")
|