nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-27 12:01:32 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

debug

Browse source
This commit is contained in:
Shen-Chenhui 2024-03-29 10:31:58 +08:00
parent d1e57cbfd5
commit 6cb4e83d8a
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
opensora/models/vae/vae_3d.py
View file

@ -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,