zhaozhilin
commited on
Commit
·
6f2f3b5
1
Parent(s):
f77be9b
update
Browse files- pipeline.py +6 -6
pipeline.py
CHANGED
@@ -592,9 +592,9 @@ class SDXLLongPromptWeightingPipeline(
|
|
592 |
model_cpu_offload_seq = "text_encoder->text_encoder_2->image_encoder->unet->vae"
|
593 |
_optional_components = [
|
594 |
"tokenizer",
|
595 |
-
"tokenizer_2",
|
596 |
"text_encoder",
|
597 |
-
"text_encoder_2",
|
598 |
"image_encoder",
|
599 |
"feature_extractor",
|
600 |
]
|
@@ -612,9 +612,9 @@ class SDXLLongPromptWeightingPipeline(
|
|
612 |
self,
|
613 |
vae: AutoencoderKL,
|
614 |
text_encoder: CLIPTextModel,
|
615 |
-
text_encoder_2: CLIPTextModelWithProjection,
|
616 |
tokenizer: CLIPTokenizer,
|
617 |
-
tokenizer_2: CLIPTokenizer,
|
618 |
unet: UNet2DConditionModel,
|
619 |
scheduler: KarrasDiffusionSchedulers,
|
620 |
feature_extractor: Optional[CLIPImageProcessor] = None,
|
@@ -627,9 +627,9 @@ class SDXLLongPromptWeightingPipeline(
|
|
627 |
self.register_modules(
|
628 |
vae=vae,
|
629 |
text_encoder=text_encoder,
|
630 |
-
text_encoder_2=
|
631 |
tokenizer=tokenizer,
|
632 |
-
tokenizer_2=
|
633 |
unet=unet,
|
634 |
scheduler=scheduler,
|
635 |
feature_extractor=feature_extractor,
|
|
|
592 |
model_cpu_offload_seq = "text_encoder->text_encoder_2->image_encoder->unet->vae"
|
593 |
_optional_components = [
|
594 |
"tokenizer",
|
595 |
+
# "tokenizer_2",
|
596 |
"text_encoder",
|
597 |
+
# "text_encoder_2",
|
598 |
"image_encoder",
|
599 |
"feature_extractor",
|
600 |
]
|
|
|
612 |
self,
|
613 |
vae: AutoencoderKL,
|
614 |
text_encoder: CLIPTextModel,
|
615 |
+
# text_encoder_2: CLIPTextModelWithProjection,
|
616 |
tokenizer: CLIPTokenizer,
|
617 |
+
# tokenizer_2: CLIPTokenizer,
|
618 |
unet: UNet2DConditionModel,
|
619 |
scheduler: KarrasDiffusionSchedulers,
|
620 |
feature_extractor: Optional[CLIPImageProcessor] = None,
|
|
|
627 |
self.register_modules(
|
628 |
vae=vae,
|
629 |
text_encoder=text_encoder,
|
630 |
+
text_encoder_2=text_encoder,
|
631 |
tokenizer=tokenizer,
|
632 |
+
tokenizer_2=tokenizer,
|
633 |
unet=unet,
|
634 |
scheduler=scheduler,
|
635 |
feature_extractor=feature_extractor,
|