Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
5ded26cd
Unverified
Commit
5ded26cd
authored
Apr 07, 2025
by
alex choi
Committed by
GitHub
Apr 07, 2025
Browse files
ensure dtype match between diffused latents and vae weights (#8391)
parent
506f39af
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
src/diffusers/pipelines/pixart_alpha/pipeline_pixart_sigma.py
...diffusers/pipelines/pixart_alpha/pipeline_pixart_sigma.py
+1
-1
No files found.
src/diffusers/pipelines/pixart_alpha/pipeline_pixart_sigma.py
View file @
5ded26cd
...
...
@@ -868,7 +868,7 @@ class PixArtSigmaPipeline(DiffusionPipeline):
xm
.
mark_step
()
if
not
output_type
==
"latent"
:
image
=
self
.
vae
.
decode
(
latents
/
self
.
vae
.
config
.
scaling_factor
,
return_dict
=
False
)[
0
]
image
=
self
.
vae
.
decode
(
latents
.
to
(
self
.
vae
.
dtype
)
/
self
.
vae
.
config
.
scaling_factor
,
return_dict
=
False
)[
0
]
if
use_resolution_binning
:
image
=
self
.
image_processor
.
resize_and_crop_tensor
(
image
,
orig_width
,
orig_height
)
else
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment