From 721fe5859d371c8d61b247b3984b880d0ef08208 Mon Sep 17 00:00:00 2001
From: b_rookie <guoboqiang@huawei.com>
Date: Fri, 15 Aug 2025 10:08:38 +0800
Subject: [PATCH 1/2] fix

---
 mindscience/models/diffuser/__init__.py    | 4 ++--
 mindscience/models/transformer/__init__.py | 4 ++--
 tests/models/transformer/test_attention.py | 2 +-
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/mindscience/models/diffuser/__init__.py b/mindscience/models/diffuser/__init__.py
index 5ae956176..b87f7c937 100644
--- a/mindscience/models/diffuser/__init__.py
+++ b/mindscience/models/diffuser/__init__.py
@@ -15,7 +15,7 @@
 """
 init
 """
-from .diffusion import DDPMPipeline, DDPMScheduler, DDIMPipeline, DDIMScheduler, DiffusionTrainer
+from .diffusion import DDPMPipeline, DiffusionScheduler, DDPMScheduler, DDIMPipeline, DDIMScheduler, DiffusionTrainer
 from .diffusion_transformer import DiffusionTransformer, ConditionDiffusionTransformer
 
-__all__ = ["DDPMPipeline", "DDPMScheduler", "DDIMPipeline", "DDIMScheduler", "DiffusionTrainer", "DiffusionTransformer", "ConditionDiffusionTransformer"]
\ No newline at end of file
+__all__ = ["DiffusionScheduler", "DDPMPipeline", "DDPMScheduler", "DDIMPipeline", "DDIMScheduler", "DiffusionTrainer", "DiffusionTransformer", "ConditionDiffusionTransformer"]
\ No newline at end of file
diff --git a/mindscience/models/transformer/__init__.py b/mindscience/models/transformer/__init__.py
index eab6d08d5..b7300a82d 100644
--- a/mindscience/models/transformer/__init__.py
+++ b/mindscience/models/transformer/__init__.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 # ============================================================================
 """init"""
-from .attention import MultiHeadAttention, TransformerBlock
+from .attention import Attention, MultiHeadAttention, TransformerBlock
 from .vit import ViT
 
-__all__ = ["MultiHeadAttention", "TransformerBlock", "ViT"]
+__all__ = ["Attention", "MultiHeadAttention", "TransformerBlock", "ViT"]
diff --git a/tests/models/transformer/test_attention.py b/tests/models/transformer/test_attention.py
index 60ed1d3b4..637bbfd67 100644
--- a/tests/models/transformer/test_attention.py
+++ b/tests/models/transformer/test_attention.py
@@ -22,7 +22,7 @@ from mindspore import Tensor, ops, load_checkpoint, load_param_into_net, jit_cla
 from mindspore import dtype as mstype
 
 from mindscience.models import Attention, MultiHeadAttention, TransformerBlock, DropPath, ViT
-from mindscience.models import RelativeRMSELoss
+from mindscience.common import RelativeRMSELoss
 
 PROJECT_ROOT = os.path.abspath(os.path.join(
     os.path.dirname(__file__), "../../"))
-- 
Gitee


From b3ca0f41ac4ffce09913326a9aa80550d528ca61 Mon Sep 17 00:00:00 2001
From: Yuheng Wang <wangyuheng16@huawei.com>
Date: Fri, 15 Aug 2025 14:21:35 +0800
Subject: [PATCH 2/2] MindSPONGE init + whl name modification

---
 MindSPONGE/__init__.py | 17 +++++++++++++++++
 cmake/package.cmake    |  6 +++---
 2 files changed, 20 insertions(+), 3 deletions(-)
 create mode 100644 MindSPONGE/__init__.py

diff --git a/MindSPONGE/__init__.py b/MindSPONGE/__init__.py
new file mode 100644
index 000000000..293df70fa
--- /dev/null
+++ b/MindSPONGE/__init__.py
@@ -0,0 +1,17 @@
+# Copyright 2025 Huawei Technologies Co., Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+"""
+init
+"""
diff --git a/cmake/package.cmake b/cmake/package.cmake
index c85bcb59a..0671b4c13 100644
--- a/cmake/package.cmake
+++ b/cmake/package.cmake
@@ -9,11 +9,11 @@ set(CPACK_TEMPORARY_PACKAGE_FILE_NAME ${CMAKE_SOURCE_DIR}/build/package/mindscie
 set(CPACK_TEMPORARY_INSTALL_DIRECTORY ${CMAKE_SOURCE_DIR}/build/package/mindscience)
 
 if(ENABLE_D)
-    set(CPACK_MS_PACKAGE_NAME "mindscience_ascend")
+    set(CPACK_MS_PACKAGE_NAME "mindscience")
 elseif(ENABLE_GPU)
     set(CPACK_MS_PACKAGE_NAME "mindscience_gpu")
 else()
-    set(CPACK_MS_PACKAGE_NAME "mindscience_ascend")
+    set(CPACK_MS_PACKAGE_NAME "mindscience")
 endif()
 include(CPack)
 
@@ -23,7 +23,7 @@ set(INSTALL_PY_DIR ".")
 # copy python files
 install(
         FILES
-	    ${CMAKE_SOURCE_DIR}/mindscience/__init__.py
+        ${CMAKE_SOURCE_DIR}/mindscience/__init__.py
             ${CMAKE_SOURCE_DIR}/setup.py
         DESTINATION ${INSTALL_PY_DIR}
         COMPONENT mindscience
-- 
Gitee