"...composable_kernel_onnx.git" did not exist on "84b36bc18dd605d2e5f6393f6ede413eaa7214ed"
Commit 61b1f677 authored by comfyanonymous's avatar comfyanonymous
Browse files

Support models without previews.

parent 43d4935a
class LatentFormat: class LatentFormat:
scale_factor = 1.0
latent_rgb_factors = None
taesd_decoder_name = None
def process_in(self, latent): def process_in(self, latent):
return latent * self.scale_factor return latent * self.scale_factor
......
...@@ -53,7 +53,9 @@ def get_previewer(device, latent_format): ...@@ -53,7 +53,9 @@ def get_previewer(device, latent_format):
method = args.preview_method method = args.preview_method
if method != LatentPreviewMethod.NoPreviews: if method != LatentPreviewMethod.NoPreviews:
# TODO previewer methods # TODO previewer methods
taesd_decoder_path = folder_paths.get_full_path("vae_approx", latent_format.taesd_decoder_name) taesd_decoder_path = None
if latent_format.taesd_decoder_name is not None:
taesd_decoder_path = folder_paths.get_full_path("vae_approx", latent_format.taesd_decoder_name)
if method == LatentPreviewMethod.Auto: if method == LatentPreviewMethod.Auto:
method = LatentPreviewMethod.Latent2RGB method = LatentPreviewMethod.Latent2RGB
...@@ -68,7 +70,8 @@ def get_previewer(device, latent_format): ...@@ -68,7 +70,8 @@ def get_previewer(device, latent_format):
print("Warning: TAESD previews enabled, but could not find models/vae_approx/{}".format(latent_format.taesd_decoder_name)) print("Warning: TAESD previews enabled, but could not find models/vae_approx/{}".format(latent_format.taesd_decoder_name))
if previewer is None: if previewer is None:
previewer = Latent2RGBPreviewer(latent_format.latent_rgb_factors) if latent_format.latent_rgb_factors is not None:
previewer = Latent2RGBPreviewer(latent_format.latent_rgb_factors)
return previewer return previewer
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment