Vision-CAIR commited on
Commit
131d43b
β€’
1 Parent(s): f5a2669

Update minigpt4/models/mini_gpt4.py

Browse files
Files changed (1) hide show
  1. minigpt4/models/mini_gpt4.py +1 -1
minigpt4/models/mini_gpt4.py CHANGED
@@ -96,7 +96,7 @@ class MiniGPT4(Blip2Base):
96
  )
97
  else:
98
  self.llama_model = LlamaForCausalLM.from_pretrained(
99
- 'Vision-CAIR/vicuna-b', load_in_8bit=True, torch_dtype=torch.float16, device_map="auto", use_auth_token=os.environ["API_TOKEN"]
100
  )
101
  for name, param in self.llama_model.named_parameters():
102
  param.requires_grad = False
 
96
  )
97
  else:
98
  self.llama_model = LlamaForCausalLM.from_pretrained(
99
+ 'Vision-CAIR/vicuna-7b', load_in_8bit=True, torch_dtype=torch.float16, device_map="auto", use_auth_token=os.environ["API_TOKEN"]
100
  )
101
  for name, param in self.llama_model.named_parameters():
102
  param.requires_grad = False