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
622f35b1
Unverified
Commit
622f35b1
authored
Sep 28, 2023
by
Ömer Veysel Çağatan
Committed by
GitHub
Sep 28, 2023
Browse files
fixed vae scaling (#5213)
parent
39baf0b4
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
0 deletions
+2
-0
src/diffusers/pipelines/latent_diffusion_uncond/pipeline_latent_diffusion_uncond.py
...tent_diffusion_uncond/pipeline_latent_diffusion_uncond.py
+2
-0
No files found.
src/diffusers/pipelines/latent_diffusion_uncond/pipeline_latent_diffusion_uncond.py
View file @
622f35b1
...
...
@@ -114,6 +114,8 @@ class LDMPipeline(DiffusionPipeline):
# compute the previous noisy sample x_t -> x_t-1
latents
=
self
.
scheduler
.
step
(
noise_prediction
,
t
,
latents
,
**
extra_kwargs
).
prev_sample
# adjust latents with inverse of vae scale
latents
=
latents
/
self
.
vqvae
.
config
.
scaling_factor
# decode the image latents with the VAE
image
=
self
.
vqvae
.
decode
(
latents
).
sample
...
...
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