From e0728bc94bdcc1934fef1d71054725b2480a6acc Mon Sep 17 00:00:00 2001 From: chengzeyi Date: Thu, 22 Feb 2024 15:23:04 +0800 Subject: [PATCH] fix optimize_stable_video_diffusion_pipeline --- examples/optimize_stable_video_diffusion_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/optimize_stable_video_diffusion_pipeline.py b/examples/optimize_stable_video_diffusion_pipeline.py index ad8bffa..e9b3977 100644 --- a/examples/optimize_stable_video_diffusion_pipeline.py +++ b/examples/optimize_stable_video_diffusion_pipeline.py @@ -195,7 +195,7 @@ def quantize_unet(m): model.unet = torch.compile(model.unet, mode=mode) if hasattr(model, 'controlnet'): model.controlnet = torch.compile(model.controlnet, mode=mode) - # model.vae = torch.compile(model.vae, mode=mode) + model.vae = torch.compile(model.vae, mode=mode) else: raise ValueError(f'Unknown compiler: {args.compiler}')