diff --git a/diffusers_helper/k_diffusion_hunyuan.py b/diffusers_helper/k_diffusion_hunyuan.py index bbb86c7..f081452 100644 --- a/diffusers_helper/k_diffusion_hunyuan.py +++ b/diffusers_helper/k_diffusion_hunyuan.py @@ -23,7 +23,6 @@ def get_flux_sigmas_from_mu(n, mu): sigmas = flux_time_shift(sigmas, mu=mu) return sigmas - @torch.inference_mode() def sample_hunyuan2( transformer,