Spaces:
Sleeping
Sleeping
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="
|
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="",
|