From 6cb4e83d8a5cdd54adaaa4017fb7d582d1db24aa Mon Sep 17 00:00:00 2001
From: Shen-Chenhui <e0052680@u.nus.edu>
Date: Fri, 29 Mar 2024 10:31:58 +0800
Subject: [PATCH] debug

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

diff --git a/opensora/models/vae/vae_3d.py b/opensora/models/vae/vae_3d.py
index 822b038..ed88b82 100644
--- a/opensora/models/vae/vae_3d.py
+++ b/opensora/models/vae/vae_3d.py
@@ -130,7 +130,7 @@ class Encoder(nn.Module):
         self.block_args = dict(
             # norm_fn=self.norm_fn,
             conv_fn=self.conv_fn,
-            dtype=self.dtype,
+            dtype=dtype,
             activation_fn=self.activation_fn,
             use_conv_shortcut=False,
             num_groups=self.num_groups,