diff --git a/MindIE/MultiModal/OpenSora-1.2/opensora/stdit3/stdit3.py b/MindIE/MultiModal/OpenSora-1.2/opensora/stdit3/stdit3.py index 348a21df5ce506217d846308c9d7042156f05597..cb803b0a85fa5a12dd55b9da05b4958a70939847 100644 --- a/MindIE/MultiModal/OpenSora-1.2/opensora/stdit3/stdit3.py +++ b/MindIE/MultiModal/OpenSora-1.2/opensora/stdit3/stdit3.py @@ -261,7 +261,7 @@ class STDiT3Block(nn.Module): class STDiT3(DiffusionModel): config_class = STDiT3Config - weigths_name = 'model.safetensors' + weights_name = 'model.safetensors' def __init__(self, config): super().__init__(config) diff --git a/MindIE/MultiModal/OpenSora-1.2/opensora/vae/VideoAutoencoder.py b/MindIE/MultiModal/OpenSora-1.2/opensora/vae/VideoAutoencoder.py index 4bffccdd030e88a2d0bccb426408d6a41b720c66..96101bab19a2d492baa5f25603380e294a0ff933 100644 --- a/MindIE/MultiModal/OpenSora-1.2/opensora/vae/VideoAutoencoder.py +++ b/MindIE/MultiModal/OpenSora-1.2/opensora/vae/VideoAutoencoder.py @@ -113,7 +113,7 @@ class VideoAutoencoderKL(nn.Module): class VideoAutoencoder(DiffusionModel): config_class = VideoAutoencoderConfig - weigths_name = 'model.safetensors' + weights_name = 'model.safetensors' def __init__(self, config: VideoAutoencoderConfig): super().__init__(config=config)