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

fix module names in comments

Browse Source
This commit is contained in:
Lincoln Stein
2023-04-08 11:07:15 -04:00
committed by GitHub
parent 307cfc075d
commit fe3f9d41fc
1 changed files with 2 additions and 2 deletions
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()