Update ttl_nn_latent_upscaler.py

This commit is contained in:
VALADI K JAGANATHAN
2023-11-30 14:51:52 +05:30
committed by GitHub
parent b22f09ad5a
commit 0c70bf32c4

View File

@@ -310,4 +310,12 @@ class NNLatentUpscale:
latent_out = latent_out.to(device="cpu")
self.model.to(device=model_management.vae_offload_device())
return ({"samples": latent_out},)
return ({"samples": latent_out},)
NODE_CLASS_MAPPINGS = {
"NNLatentUpscale": NNLatentUpscale
}
NODE_DISPLAY_NAME_MAPPINGS = {
"NNlLatentUpscale": "EFF-NN Latent Upscale"
}