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

change to cuda

Browse Source
This commit is contained in:
3luka 2024-08-16 13:35:52 +03:00
parent 79ead58faf
commit b6d44705b9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
minigpt4/models/minigpt4.py
View File

@ -189,7 +189,7 @@ class MiniGPT4(MiniGPTBase):
ckpt_path = cfg.get("ckpt", "") # load weights of MiniGPT-4 ckpt_path = cfg.get("ckpt", "") # load weights of MiniGPT-4
if ckpt_path: if ckpt_path:
print("Load MiniGPT-4 Checkpoint: {}".format(ckpt_path)) print("Load MiniGPT-4 Checkpoint: {}".format(ckpt_path))
ckpt = torch.load(ckpt_path, map_location="cpu") ckpt = torch.load(ckpt_path, map_location="cuda")
msg = model.load_state_dict(ckpt['model'], strict=False) msg = model.load_state_dict(ckpt['model'], strict=False)
return model return model