Upload pipeline_interactdiffusion_sdxl.py with huggingface_hub
Browse files
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:
|
| 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,
|