update

bubogpt/models/mm_gpt4.py

@@ -87,9 +87,8 @@ class MMGPT4(BaseModel):
         self.llama_tokenizer = LlamaTokenizer.from_pretrained('magicr/vicuna-7b', use_fast=False, use_auth_token=True)
         self.llama_tokenizer.pad_token = self.llama_tokenizer.eos_token
 
-        self.llama_model = LlamaForCausalLM.from_pretrained(
-            'magicr/vicuna-7b', load_in_8bit=True, torch_dtype=torch.float16, device_map="auto", use_auth_token=True
-        )
+        self.llama_model = LlamaForCausalLM.from_pretrained('magicr/vicuna-7b', load_in_8bit=True,
+                                                            torch_dtype=torch.float16, device_map="auto", use_auth_token=True)
 
         if freeze_llm:
             for name, param in self.llama_model.named_parameters():