Machine-Learning
/
aiuNN
1
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

feat/smoe_support #6

Merged
Fabel merged 4 commits from feat/smoe_support into develop 2025-03-27 16:19:00 +00:00
Conversation 0 Commits 4 Files Changed 5 +4 -3
3 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 5c668e3c7b - Show all commits

2
setup.py
View File

@ -2,7 +2,7 @@ from setuptools import setup, find_packages
setup( setup(
name="aiunn", name="aiunn",
version="0.1.1", version="0.1.2",
packages=find_packages(where="src"), packages=find_packages(where="src"),
package_dir={"": "src"}, package_dir={"": "src"},
install_requires=[ install_requires=[

2
src/aiunn/__init__.py
View File

@ -3,4 +3,4 @@ from .upsampler.aiunn import aiuNN
from .upsampler.config import aiuNNConfig from .upsampler.config import aiuNNConfig
from .inference.inference import aiuNNInference from .inference.inference import aiuNNInference
__version__ = "0.1.1" __version__ = "0.1.2"

3
src/aiunn/upsampler/aiunn.py
View File

@ -2,7 +2,8 @@ import os
import torch import torch
import torch.nn as nn import torch.nn as nn
import warnings import warnings
from aiia import AIIA, AIIAConfig, AIIABase from aiia import AIIAConfig, AIIABase
from aiia.model.Model import AIIA
from .config import aiuNNConfig from .config import aiuNNConfig
import warnings import warnings