Upload modeling_mplug_owl2.py with huggingface_hub

modeling_mplug_owl2.py

@@ -261,7 +261,6 @@ class MPLUGOwl2LlamaForCausalLM(LlamaForCausalLM, MPLUGOwl2MetaForCausalLM):
         # Initialize weights and apply final processing
         self.post_init()
         
-        self.weight_tensor = torch.Tensor([5.,4.,3.,2.,1.]).half().to(self.device)
 
     def get_model(self):
         return self.model
@@ -270,6 +269,8 @@ class MPLUGOwl2LlamaForCausalLM(LlamaForCausalLM, MPLUGOwl2MetaForCausalLM):
               task_: str = "quality",
               input_: str = "image",
              ):
+        if not hasattr(self, "weight_tensor"):
+            self.weight_tensor = torch.Tensor([5.,4.,3.,2.,1.]).half().to(self.device)
         prompt = "USER: How would you rate the {} of this {}?\n<|image|>\nASSISTANT: The {} of the {} is".format(task_, input_, input_, task_)
         if input_ == "image":
             images = [expand2square(img, tuple(int(x*255) for x in self.image_processor.image_mean)) for img in images]