diffusers
/

HunyuanVideo-vae

Model card Files Files and versions Community

hlky HF Staff commited on Mar 4

Commit

3d28980

·

verified ·

1 Parent(s): d07745d

Update handler.py

Files changed (1) hide show

handler.py +1 -0

handler.py CHANGED Viewed

@@ -11,6 +11,7 @@ class EndpointHandler:
         self.device = "cuda"
         self.dtype = torch.float16
         self.vae = cast(AutoencoderKLHunyuanVideo, AutoencoderKLHunyuanVideo.from_pretrained(path, torch_dtype=self.dtype).to(self.device, self.dtype).eval())
         self.vae_scale_factor = self.vae_scale_factor_spatial = self.vae.spatial_compression_ratio
         self.video_processor = VideoProcessor(

         self.device = "cuda"
         self.dtype = torch.float16
         self.vae = cast(AutoencoderKLHunyuanVideo, AutoencoderKLHunyuanVideo.from_pretrained(path, torch_dtype=self.dtype).to(self.device, self.dtype).eval())
+        self.vae.enable_tiling()
         self.vae_scale_factor = self.vae_scale_factor_spatial = self.vae.spatial_compression_ratio
         self.video_processor = VideoProcessor(