Merge pull request #44 from jags111/jags111-patch-1

Update ttl_nn_latent_upscaler.py
This commit is contained in:
VALADI K JAGANATHAN
2023-11-30 14:57:14 +05:30
committed by GitHub

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"
}