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,
|