From 1f219925576fda48cd1d8f606f6c3c17068b425f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=BE=99=E9=92=A2?= Date: Tue, 1 Jul 2025 17:57:16 +0800 Subject: [PATCH] weights_name bugfix --- MindIE/MultiModal/OpenSora-1.2/opensora/stdit3/stdit3.py | 2 +- MindIE/MultiModal/OpenSora-1.2/opensora/vae/VideoAutoencoder.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/MindIE/MultiModal/OpenSora-1.2/opensora/stdit3/stdit3.py b/MindIE/MultiModal/OpenSora-1.2/opensora/stdit3/stdit3.py index 348a21df5c..cb803b0a85 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 4bffccdd03..96101bab19 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) -- Gitee