From 86f3a80f51eb0ca4f61c34297bffa3a292817244 Mon Sep 17 00:00:00 2001 From: Ean Garvey Date: Tue, 9 Jul 2024 15:15:50 -0500 Subject: [PATCH] Add TODO to encode comment in VAE export script. --- models/turbine_models/custom_models/sd_inference/vae.py | 1 + 1 file changed, 1 insertion(+) diff --git a/models/turbine_models/custom_models/sd_inference/vae.py b/models/turbine_models/custom_models/sd_inference/vae.py index 9645ebfdc..6fdbfd958 100644 --- a/models/turbine_models/custom_models/sd_inference/vae.py +++ b/models/turbine_models/custom_models/sd_inference/vae.py @@ -183,6 +183,7 @@ def export_vae_model( ): fxb = FxProgramsBuilder(vae_model) + # TODO: fix issues with exporting the encode function. # @fxb.export_program(args=(encode_args,)) # def _encode(module, inputs,): # return module.encode(*inputs)