kiigii commited on
Commit
18ca1e8
·
verified ·
1 Parent(s): 59cc439

set num_views as attr of attn_processor to support torch.compile

Browse files
Files changed (1) hide show
  1. pipeline_imagedream.py +1 -6
pipeline_imagedream.py CHANGED
@@ -32,7 +32,7 @@ from diffusers.pipelines.stable_diffusion.safety_checker import (
32
  StableDiffusionSafetyChecker,
33
  )
34
  from diffusers.schedulers import KarrasDiffusionSchedulers
35
- from diffusers.utils import deprecate, logging
36
  from transformers import (
37
  CLIPImageProcessor,
38
  CLIPTextModel,
@@ -90,11 +90,6 @@ class ImageDreamPipeline(StableDiffusionPipeline):
90
  setattr(self.image_encoder, "visual_projection", nn.Identity())
91
  add_imagedream_attn_processor(self.unet)
92
  set_num_views(self.unet, self.num_views + 1)
93
- logging.set_verbosity_error()
94
- print(
95
- "ImageDream Cross-Attention uses `num_views` kwarg, "
96
- "and set logging verbosity to error."
97
- )
98
 
99
  def unload_ip_adapter(self) -> None:
100
  super().unload_ip_adapter()
 
32
  StableDiffusionSafetyChecker,
33
  )
34
  from diffusers.schedulers import KarrasDiffusionSchedulers
35
+ from diffusers.utils import deprecate
36
  from transformers import (
37
  CLIPImageProcessor,
38
  CLIPTextModel,
 
90
  setattr(self.image_encoder, "visual_projection", nn.Identity())
91
  add_imagedream_attn_processor(self.unet)
92
  set_num_views(self.unet, self.num_views + 1)
 
 
 
 
 
93
 
94
  def unload_ip_adapter(self) -> None:
95
  super().unload_ip_adapter()