diff --git a/comfy_extras/nodes_latent.py b/comfy_extras/nodes_latent.py index 362a83caa..8f2ee40df 100644 --- a/comfy_extras/nodes_latent.py +++ b/comfy_extras/nodes_latent.py @@ -1,4 +1,5 @@ import torch +from nodes import MAX_RESOLUTION class NoisyLatentImage: def __init__(self, device="cpu"):