perling
/

diffusers_pipeline

Model card Files Files and versions

zhaozhilin commited on Jan 10, 2024

Commit

9629b0c

·

1 Parent(s): ecdc6c7

update

Files changed (1) hide show

pipeline.py +2 -2

pipeline.py CHANGED Viewed

@@ -491,7 +491,7 @@ class OnnxStableDiffusionLongPromptWeightingPipeline(OnnxStableDiffusionPipeline
         self.unet.config.in_channels = 4
         self.vae_scale_factor = 8
-    def _encode_prompt(
         self,
         prompt,
         num_images_per_prompt,
@@ -764,7 +764,7 @@ class OnnxStableDiffusionLongPromptWeightingPipeline(OnnxStableDiffusionPipeline
         do_classifier_free_guidance = guidance_scale > 1.0
         # 3. Encode input prompt
-        text_embeddings = self._encode_prompt(
             prompt,
             num_images_per_prompt,
             do_classifier_free_guidance,

         self.unet.config.in_channels = 4
         self.vae_scale_factor = 8
+    def encode_prompt(
         self,
         prompt,
         num_images_per_prompt,
         do_classifier_free_guidance = guidance_scale > 1.0
         # 3. Encode input prompt
+        text_embeddings = self.encode_prompt(
             prompt,
             num_images_per_prompt,
             do_classifier_free_guidance,