Update city96_latent_upscaler.py

This commit is contained in:
VALADI K JAGANATHAN
2023-11-30 14:50:21 +05:30
committed by GitHub
parent 9090556dcd
commit b22f09ad5a

View File

@@ -85,4 +85,10 @@ class LatentUpscaler:
mask = torch.nn.functional.interpolate(samples['noise_mask'], scale_factor=float(scale_factor), mode='bicubic')
return ({"samples": lt, "noise_mask": mask},)
return ({"samples": lt},)
NODE_CLASS_MAPPINGS = {
"LatentUpscaler": LatentUpscaler,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"LatentUpscaler": "EFF-Latent Upscaler"
}