zhaozhilin commited on
Commit
6f2f3b5
·
1 Parent(s): f77be9b
Files changed (1) hide show
  1. 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=text_encoder_2,
631
  tokenizer=tokenizer,
632
- tokenizer_2=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,