From 9ee27edd1855a1f6e9f8d7e053f48dd3349671d4 Mon Sep 17 00:00:00 2001 From: xingzhongfan Date: Mon, 21 Apr 2025 16:31:04 +0800 Subject: [PATCH] update sno init --- MindFlow/mindflow/cell/__init__.py | 4 ++-- MindFlow/mindflow/cell/neural_operators/__init__.py | 5 +++-- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/MindFlow/mindflow/cell/__init__.py b/MindFlow/mindflow/cell/__init__.py index 80bb71339..59bbc9634 100644 --- a/MindFlow/mindflow/cell/__init__.py +++ b/MindFlow/mindflow/cell/__init__.py @@ -15,7 +15,7 @@ """init""" from .activation import get_activation from .basic_block import LinearBlock, ResBlock, InputScale, FCSequential, MultiScaleFCSequential, DropPath -from .neural_operators import FNO1D, FNO2D, FNO3D, KNO1D, KNO2D, PDENet, PeRCNN, SNO1D, SNO2D, SNO3D +from .neural_operators import FNO1D, FNO2D, FNO3D, KNO1D, KNO2D, PDENet, PeRCNN, SNO, SNO1D, SNO2D, SNO3D from .attention import Attention, MultiHeadAttention, AttentionBlock from .vit import ViT from .unet2d import UNet2D @@ -24,7 +24,7 @@ from .diffusion import DiffusionScheduler, DiffusionTrainer, DDPMScheduler, DDIM from .diffusion_transformer import DiffusionTransformer, ConditionDiffusionTransformer __all__ = ["get_activation", "FNO1D", "FNO2D", "FNO3D", "KNO1D", "KNO2D", "PDENet", "UNet2D", "PeRCNN", - "SNO1D", "SNO2D", "SNO3D", "Attention", "MultiHeadAttention", "AttentionBlock", "ViT", "DDPMPipeline", + "SNO", "SNO1D", "SNO2D", "SNO3D", "Attention", "MultiHeadAttention", "AttentionBlock", "ViT", "DDPMPipeline", "DDIMPipeline", "DiffusionTrainer", "DiffusionScheduler", "DDPMScheduler", "DDIMScheduler", "DiffusionTransformer", "ConditionDiffusionTransformer"] __all__.extend(basic_block.__all__) diff --git a/MindFlow/mindflow/cell/neural_operators/__init__.py b/MindFlow/mindflow/cell/neural_operators/__init__.py index 60d482175..4498dba83 100644 --- a/MindFlow/mindflow/cell/neural_operators/__init__.py +++ b/MindFlow/mindflow/cell/neural_operators/__init__.py @@ -18,8 +18,9 @@ from .kno1d import KNO1D from .kno2d import KNO2D from .pdenet import PDENet from .percnn import PeRCNN -from .sno import SNO1D, SNO2D, SNO3D +from .sno import SNO, SNO1D, SNO2D, SNO3D -__all__ = ["FNOBlocks", "FNO1D", "FNO2D", "FNO3D", "KNO1D", "KNO2D", "PDENet", "PeRCNN", "SNO1D", "SNO2D", "SNO3D"] +__all__ = ["FNOBlocks", "FNO1D", "FNO2D", "FNO3D", "KNO1D", "KNO2D", "PDENet", "PeRCNN", + "SNO", "SNO1D", "SNO2D", "SNO3D"] __all__.sort() -- Gitee