fix optimize_stable_video_diffusion_pipeline

chengzeyi · Feb 22, 2024 · e0728bc · e0728bc
1 parent 7e36d62
commit e0728bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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}')