revert some modifications

pull/1685/head
mzr1996 2023-07-21 17:17:30 +08:00
parent bb35d1fab9
commit 807ce327db
3 changed files with 2 additions and 12 deletions

View File

@ -52,13 +52,8 @@ model = dict(
'https://download.openmmlab.com/mmpretrain/v1.0/minigpt4/minigpt-4_qformer_20230615-1dfa889c.pth' # noqa
),
lang_encoder=dict(
type='AutoModelForCausalLM',
name_or_path=
'/mnt/petrelfs/share_data/liuyuan/llm_weights/vicuna_weights_7b'),
tokenizer=dict(
type='LlamaTokenizer',
name_or_path=
'/mnt/petrelfs/share_data/liuyuan/llm_weights/vicuna_weights_7b'),
type='AutoModelForCausalLM', name_or_path='YOUR_PATH_TO_VICUNA'),
tokenizer=dict(type='LlamaTokenizer', name_or_path='YOUR_PATH_TO_VICUNA'),
task='caption',
prompt_template='###Human: {} ###Assistant: ',
raw_prompts=[

Binary file not shown.

Before

Width:  |  Height:  |  Size: 45 KiB

View File

@ -1,5 +0,0 @@
from mmpretrain import inference_model
result = inference_model('instructblip-vicuna7b_3rdparty-zeroshot_caption',
'demo/cat-dog.png')
print(result)