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

fix: maybe cpu mode

Browse Source
This commit is contained in:
Vandivier 2023-04-23 15:03:55 -06:00
parent 3ca6fff115
commit fb7dd787f9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
demo.py
View File

@ -57,11 +57,11 @@ cfg = Config(args)
model_config = cfg.model_cfg model_config = cfg.model_cfg
model_config.device_8bit = args.gpu_id model_config.device_8bit = args.gpu_id
model_cls = registry.get_model_class(model_config.arch) model_cls = registry.get_model_class(model_config.arch)
model = model_cls.from_config(model_config).to('cuda:{}'.format(args.gpu_id)) model = model_cls.from_config(model_config).to('cpu')
vis_processor_cfg = cfg.datasets_cfg.cc_sbu_align.vis_processor.train vis_processor_cfg = cfg.datasets_cfg.cc_sbu_align.vis_processor.train
vis_processor = registry.get_processor_class(vis_processor_cfg.name).from_config(vis_processor_cfg) vis_processor = registry.get_processor_class(vis_processor_cfg.name).from_config(vis_processor_cfg)
chat = Chat(model, vis_processor, device='cuda:{}'.format(args.gpu_id)) chat = Chat(model, vis_processor, device='cpu')
print('Initialization Finished') print('Initialization Finished')
# ======================================== # ========================================