zhaozhilin
commited on
Commit
·
9629b0c
1
Parent(s):
ecdc6c7
update
Browse files- 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
|
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.
|
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,
|