nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-20 17:35:58 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

debug

Browse source
This commit is contained in:
Shen-Chenhui 2024-04-11 15:37:22 +08:00
parent 4cb946c547
commit f6c79a75c2
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
opensora/models/vae/vae_3d_v2.py
View file

@ -276,7 +276,9 @@ class StyleGANDiscriminator(nn.Module):
num_groups=32,
dtype = torch.bfloat16,
device="cpu",
):
):
super().__init__()
self.dtype = dtype
self.input_size = cast_tuple(image_size, 2)
self.filters = discriminator_filters