Update lavis/models/protein_models/protein_function_opt.py

lavis/models/protein_models/protein_function_opt.py

@@ -100,7 +100,7 @@ class Blip2ProteinMistral(Blip2ProteinBase):
         # self.mistral_tokenizer = LlamaTokenizer.from_pretrained("/cluster/home/wenkai/.cache/huggingface/hub/models--teknium--OpenHermes-2.5-Mistral-7B", use_fast=False)
         # configuration = MistralConfig()
         self.mistral_tokenizer.pad_token = '<pad>'
-        self.mistral_model = MistralForCausalLM.from_pretrained("teknium/OpenHermes-2.5-Mistral-7B")
+        self.mistral_model = MistralForCausalLM.from_pretrained("teknium/OpenHermes-2.5-Mistral-7B", torch_dtype=torch.float16)
         # self.mistral_model = MistralForCausalLM.from_pretrained("/cluster/home/wenkai/.cache/huggingface/hub/models--teknium--OpenHermes-2.5-Mistral-7B", torch_dtype=torch.float16)
         # self.mistral_model = MistralForCausalLM(configuration)
         for name, param in self.mistral_model.named_parameters():