zhaozhilin commited on
Commit
9629b0c
·
1 Parent(s): ecdc6c7
Files changed (1) hide show
  1. pipeline.py +2 -2
pipeline.py CHANGED
@@ -491,7 +491,7 @@ class OnnxStableDiffusionLongPromptWeightingPipeline(OnnxStableDiffusionPipeline
491
  self.unet.config.in_channels = 4
492
  self.vae_scale_factor = 8
493
 
494
- def _encode_prompt(
495
  self,
496
  prompt,
497
  num_images_per_prompt,
@@ -764,7 +764,7 @@ class OnnxStableDiffusionLongPromptWeightingPipeline(OnnxStableDiffusionPipeline
764
  do_classifier_free_guidance = guidance_scale > 1.0
765
 
766
  # 3. Encode input prompt
767
- text_embeddings = self._encode_prompt(
768
  prompt,
769
  num_images_per_prompt,
770
  do_classifier_free_guidance,
 
491
  self.unet.config.in_channels = 4
492
  self.vae_scale_factor = 8
493
 
494
+ def encode_prompt(
495
  self,
496
  prompt,
497
  num_images_per_prompt,
 
764
  do_classifier_free_guidance = guidance_scale > 1.0
765
 
766
  # 3. Encode input prompt
767
+ text_embeddings = self.encode_prompt(
768
  prompt,
769
  num_images_per_prompt,
770
  do_classifier_free_guidance,