Update pipeline_imagedream.py

pipeline_imagedream.py

@@ -59,9 +59,9 @@ class ImageDreamPipeline(StableDiffusionPipeline):
         self,
         pretrained_model_name_or_path_or_dict: Union[
             str, List[str], Dict[str, torch.Tensor]
-        ],
-        subfolder: Union[str, List[str]],
-        weight_name: Union[str, List[str]],
+        ] = "kiigii/imagedream-ipmv-diffusers",
+        subfolder: Union[str, List[str]] = "ip_adapter",
+        weight_name: Union[str, List[str]] = "ip-adapter-plus_imagedream.bin",
         image_encoder_folder: Optional[str] = "image_encoder",
         **kwargs,
     ):
@@ -72,7 +72,15 @@ class ImageDreamPipeline(StableDiffusionPipeline):
             image_encoder_folder=image_encoder_folder,
             **kwargs,
         )
-        add_imagedream_attn_processor(self.unet)
+        print("IP-Adapter Loaded.")
+
+        if pretrained_model_name_or_path_or_dict == "kiigii/imagedream-ipmv-diffusers":
+            add_imagedream_attn_processor(self.unet)
+            logging.set_verbosity_error()
+            print(
+                "ImageDream Cross-Attention uses `num_views` kwarg, "
+                "and set logging verbosity to error."
+            )
 
     def encode_image_to_latents(
         self,