saicharan1234 commited on
Commit
840411f
·
verified ·
1 Parent(s): de2fe82

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="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,
@@ -38,7 +38,7 @@ class MiniGPT4(Blip2Base):
38
  freeze_vit=True,
39
  freeze_qformer=True,
40
  num_query_token=32,
41
- llama_model="",
42
  llama_cache_dir='',
43
  prompt_path="",
44
  prompt_template="",
 
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,
 
38
  freeze_vit=True,
39
  freeze_qformer=True,
40
  num_query_token=32,
41
+ llama_model="/data/vicuna-7b",
42
  llama_cache_dir='',
43
  prompt_path="",
44
  prompt_template="",