chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-05 02:20:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix the device bug with latest transformers

Browse Source
This commit is contained in:
Deyao Zhu 2023-08-29 16:12:22 +03:00
parent 81d7314a02
commit 871918f758
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
eval_configs/minigpt4_eval.yaml
View File

@ -5,7 +5,7 @@ model:
end_sym: "###" end_sym: "###"
low_resource: True low_resource: True
prompt_template: '###Human: {} ###Assistant: ' prompt_template: '###Human: {} ###Assistant: '
ckpt: '/home/zhud/ibex/pretrained_minigpt4.pth' ckpt: '/path/to/checkpoint/'
datasets: datasets:

2
eval_configs/minigpt4_llama2_eval.yaml
View File

@ -5,7 +5,7 @@ model:
end_sym: "</s>" end_sym: "</s>"
low_resource: True low_resource: True
prompt_template: '[INST] {} [/INST] ' prompt_template: '[INST] {} [/INST] '
ckpt: '/home/zhud/c2090/zhud/project/MiniGPT-4/minigpt4/output/minigpt4_stage2_finetune/20230826182/checkpoint_4.pth' ckpt: '/path/to/checkpoint/'
datasets: datasets:

2
minigpt4/models/base_model.py
View File

@ -24,7 +24,7 @@ class BaseModel(nn.Module):
@property @property
def device(self): def device(self):
return list(self.parameters())[0].device return list(self.parameters())[-1].device
def load_checkpoint(self, url_or_filename): def load_checkpoint(self, url_or_filename):
""" """