jiuntian commited on
Commit
64c87e1
·
verified ·
1 Parent(s): 896aaac

Upload pipeline_interactdiffusion_sdxl.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. pipeline_interactdiffusion_sdxl.py +1 -3
pipeline_interactdiffusion_sdxl.py CHANGED
@@ -53,8 +53,6 @@ from diffusers.utils import (
53
  )
54
  from diffusers.utils.torch_utils import randn_tensor
55
 
56
- from interactdiffusion_xl.interactdiffusion_unet_2d_condition import InteractDiffusionUNet2DConditionModel
57
-
58
 
59
  if is_invisible_watermark_available():
60
  from diffusers.pipelines.stable_diffusion_xl.watermark import StableDiffusionXLWatermarker
@@ -135,7 +133,7 @@ class StableDiffusionXLInteractDiffusionPipeline(
135
  text_encoder_2: CLIPTextModelWithProjection,
136
  tokenizer: CLIPTokenizer,
137
  tokenizer_2: CLIPTokenizer,
138
- unet: InteractDiffusionUNet2DConditionModel,
139
  scheduler: KarrasDiffusionSchedulers,
140
  image_encoder: CLIPVisionModelWithProjection = None,
141
  feature_extractor: CLIPImageProcessor = None,
 
53
  )
54
  from diffusers.utils.torch_utils import randn_tensor
55
 
 
 
56
 
57
  if is_invisible_watermark_available():
58
  from diffusers.pipelines.stable_diffusion_xl.watermark import StableDiffusionXLWatermarker
 
133
  text_encoder_2: CLIPTextModelWithProjection,
134
  tokenizer: CLIPTokenizer,
135
  tokenizer_2: CLIPTokenizer,
136
+ unet: UNet2DConditionModel,
137
  scheduler: KarrasDiffusionSchedulers,
138
  image_encoder: CLIPVisionModelWithProjection = None,
139
  feature_extractor: CLIPImageProcessor = None,