Spaces:

Vision-CAIR
/

minigpt4

Running on A10G

Vision-CAIR commited on May 22, 2023

Commit

131d43b

•

1 Parent(s): f5a2669

Update minigpt4/models/mini_gpt4.py

Files changed (1) hide show

minigpt4/models/mini_gpt4.py CHANGED Viewed

@@ -96,7 +96,7 @@ class MiniGPT4(Blip2Base):
             )
         else:
             self.llama_model = LlamaForCausalLM.from_pretrained(
-                'Vision-CAIR/vicuna-b', load_in_8bit=True, torch_dtype=torch.float16, device_map="auto", use_auth_token=os.environ["API_TOKEN"]
             )
         for name, param in self.llama_model.named_parameters():
             param.requires_grad = False

             )
         else:
             self.llama_model = LlamaForCausalLM.from_pretrained(
+                'Vision-CAIR/vicuna-7b', load_in_8bit=True, torch_dtype=torch.float16, device_map="auto", use_auth_token=os.environ["API_TOKEN"]
             )
         for name, param in self.llama_model.named_parameters():
             param.requires_grad = False