From d80ef70020d0a330cac3f991547907a8570673ce Mon Sep 17 00:00:00 2001
From: shenchenhui <shenchenhui@shenchenhuis-MacBook-Pro.local>
Date: Thu, 28 Mar 2024 17:15:04 +0800
Subject: [PATCH] debug

---
 opensora/models/vae/vae_3d.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/opensora/models/vae/vae_3d.py b/opensora/models/vae/vae_3d.py
index e7f81c9..44ec554 100644
--- a/opensora/models/vae/vae_3d.py
+++ b/opensora/models/vae/vae_3d.py
@@ -222,6 +222,7 @@ class Decoder(nn.Module):
         self.channel_multipliers = channel_multipliers
         self.temporal_downsample = temporal_downsample
         self.num_groups = num_groups
+        self.dtype = dtype
 
         if isinstance(self.temporal_downsample, int):
             self.temporal_downsample = _get_selected_flags(