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

Rename CropLatentsInvocation -> CropLatentsCoreInvocation to prevent conflict with custom node. And other minor tidying.

Browse Source
This commit is contained in:
Ryan Dick 2023-11-27 12:05:55 -05:00 committed by Kent Keirsey
parent 7cab51745b
commit 9b863fb9bc
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1174,14 +1174,18 @@ class BlendLatentsInvocation(BaseInvocation):
return build_latents_output(latents_name=name, latents=blended_latents) return build_latents_output(latents_name=name, latents=blended_latents)
# The Crop Latents node was copied from @skunkworxdark's implementation here:
# https://github.com/skunkworxdark/XYGrid_nodes/blob/74647fa9c1fa57d317a94bd43ca689af7f0aae5e/images_to_grids.py#L1117C1-L1167C80
@invocation( @invocation(
"lcrop", "crop_latents",
title="Crop Latents", title="Crop Latents",
tags=["latents", "crop"], tags=["latents", "crop"],
category="latents", category="latents",
version="1.0.0", version="1.0.0",
) )
class CropLatentsInvocation(BaseInvocation): # TODO(ryand): Named `CropLatentsCoreInvocation` to prevent a conflict with custom node `CropLatentsInvocation`.
# Currently, if the class names conflict then 'GET /openapi.json' fails.
class CropLatentsCoreInvocation(BaseInvocation):
"""Crops a latent-space tensor to a box specified in image-space. The box dimensions and coordinates must be """Crops a latent-space tensor to a box specified in image-space. The box dimensions and coordinates must be
divisible by the latent scale factor of 8. divisible by the latent scale factor of 8.
""" """
@ -1219,9 +1223,7 @@ class CropLatentsInvocation(BaseInvocation):
x2 = x1 + (self.width // LATENT_SCALE_FACTOR) x2 = x1 + (self.width // LATENT_SCALE_FACTOR)
y2 = y1 + (self.height // LATENT_SCALE_FACTOR) y2 = y1 + (self.height // LATENT_SCALE_FACTOR)
cropped_latents = latents[:, :, y1:y2, x1:x2] cropped_latents = latents[..., y1:y2, x1:x2]
# resized_latents = resized_latents.to("cpu")
name = f"{context.graph_execution_state_id}__{self.id}" name = f"{context.graph_execution_state_id}__{self.id}"
context.services.latents.save(name, cropped_latents) context.services.latents.save(name, cropped_latents)