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

fixed checkpoint (#92)

Browse source
This commit is contained in:
Frank Lee 2024-05-07 18:34:22 +08:00 committed by GitHub
parent fe19568a49
commit 82f15b30f9
1 changed files with 0 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
opensora/models/stdit/stdit3.py
View file

@ -164,7 +164,6 @@ class STDiT3Config(PretrainedConfig):
drop_path=0.0,
caption_channels=4096,
model_max_length=300,
dtype=torch.float32,
qk_norm=False,
enable_flash_attn=False,
enable_layernorm_kernel=False,
@ -185,7 +184,6 @@ class STDiT3Config(PretrainedConfig):
self.drop_path = drop_path
self.caption_channels = caption_channels
self.model_max_length = model_max_length
self.dtype = dtype
self.qk_norm = qk_norm
self.enable_flash_attn = enable_flash_attn
self.enable_layernorm_kernel = enable_layernorm_kernel