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

@@ -311,3 +311,11 @@ class NNLatentUpscale:
self.model.to(device=model_management.vae_offload_device()) 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"
}