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

Update minigpt4.py

Browse Source
This commit is contained in:
3luka 2024-08-16 13:28:08 +03:00
parent 18d7957e92
commit 79ead58faf
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

@ -39,7 +39,7 @@ class MiniGPT4(MiniGPTBase):
prompt_template="", prompt_template="",
max_txt_len=32, max_txt_len=32,
end_sym='\n', end_sym='\n',
low_resource=False, # use 8 bit and put vit in cpu low_resource=True, # use 8 bit and put vit in cpu
device_8bit=0, # the device of 8bit model should be set when loading and cannot be changed anymore. device_8bit=0, # the device of 8bit model should be set when loading and cannot be changed anymore.
): ):
super().__init__( super().__init__(