Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/ktransformers.git" did not exist on "a148da2cfe4706745147de1e315972a19408f6ec"
Commit cc819010 authored by patil-suraj's avatar patil-suraj
Browse files

process image

parent 7ac909d6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
models/vision/latent_diffusion/modeling_latent_diffusion.py
View file @ cc819010
...@@ -961,5 +961,6 @@ class LatentDiffusion(DiffusionPipeline): ...@@ -961,5 +961,6 @@ class LatentDiffusion(DiffusionPipeline):
image = 1 / 0.18215 * image image = 1 / 0.18215 * image
image = self.vqvae.decode(image) image = self.vqvae.decode(image)
image = torch.clamp((image+1.0)/2.0, min=0.0, max=1.0) image = torch.clamp((image+1.0)/2.0, min=0.0, max=1.0)
image = 255. * image
return image return image
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment