Update modelling_magiv2.py

modelling_magiv2.py

@@ -40,6 +40,7 @@ class Magiv2Model(PreTrainedModel):
         images = [np.array(image.convert("RGB")) for image in images]
         images = self.processor(images, return_tensors="pt").pixel_values
         images = move_to_device_fn(images)
+        images = images.to(self.dtype)
 
         # temporarily change the mask ratio from default to the one specified
         old_mask_ratio = self.crop_embedding_model.embeddings.config.mask_ratio