Update pipeline_imagedream.py

pipeline_imagedream.py

@@ -1,6 +1,7 @@
 from typing import Any, Callable, Dict, List, Optional, Union
 
 import torch
+import torch.nn as nn
 import torch.nn.functional as F
 from diffusers.callbacks import MultiPipelineCallbacks, PipelineCallback
 from diffusers.image_processor import PipelineImageInput
@@ -75,6 +76,7 @@ class ImageDreamPipeline(StableDiffusionPipeline):
         print("IP-Adapter Loaded.")
 
         if pretrained_model_name_or_path_or_dict == "kiigii/imagedream-ipmv-diffusers":
+            setattr(self.image_encoder, "visual_projection", nn.Identity())
             add_imagedream_attn_processor(self.unet)
             logging.set_verbosity_error()
             print(