Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'enhance/latents-handling' of github.com:invoke-ai/InvokeAI into enhance/latents-handling

Browse Source
This commit is contained in:
Lincoln Stein 2023-04-09 13:24:01 -04:00
commit 87798cc8fa
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/app/invocations/latent.py
View File

@ -320,7 +320,7 @@ class LatentsToImageInvocation(BaseInvocation):
image=ImageField(image_type=image_type, image_name=image_name)
)
# this should be refactored - duplicated code in diffusers.pipelines.stable_diffusion_pipeline
# this should be refactored - duplicated code in diffusers.pipelines.stable_diffusion
@classmethod
def _decode_latents(self, vae:AutoencoderKL, latents:torch.Tensor)->Image:
latents = 1 / vae.config.scaling_factor * latents
@ -353,7 +353,7 @@ class ImageToLatentsInvocation(BaseInvocation):
latents=LatentsField(latents_name=name)
)
# this should be refactored - similar code in invokeai.stable_diffusion.diffusers_pipeline
# this should be refactored - similar code in invokeai.backend.stable_diffusion.diffusers_pipeline
@classmethod
def _encode_latents(self, vae:AutoencoderKL, image:Image)->torch.Tensor:
device = choose_torch_device()