Vision-CAIR commited on
Commit
b104280
β€’
1 Parent(s): 6feb8b1

Update minigpt4/models/mini_gpt4.py

Browse files
Files changed (1) hide show
  1. minigpt4/models/mini_gpt4.py +2 -2
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")