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

Compare commits

base: Machine-Learning:25f4225baf1f5b80c00b1f7c099e79787d00dbf8
Branches Tags
Machine-Learning:main
Machine-Learning:feat/tf_support
..
compare: Machine-Learning:68a27f00c121f9e84e8c4f6de2fc8305efc5fcc5
Branches Tags
Machine-Learning:feat/tf_support
Machine-Learning:main

3 Commits
25f4225baf ... 68a27f00c1

Author SHA1 Message Date
Falko Victor Habel 68a27f00c1 included requirement for aiia 2025-03-26 22:02:11 +01:00
Falko Victor Habel 399a7c0f69 increamentet release version 2025-03-26 22:01:59 +01:00
Falko Victor Habel 5c668e3c7b updated to support new SMoe Version of AIIA 2025-03-26 22:01:47 +01:00
5 changed files with 7 additions and 5 deletions
Show Stats Expand all files Collapse all files

2
pyproject.toml
View File

@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
[project] [project]
name = "aiunn" name = "aiunn"
version = "0.1.1" version = "0.1.2"
description = "Finetuner for image upscaling using AIIA" description = "Finetuner for image upscaling using AIIA"
readme = "README.md" readme = "README.md"
requires-python = ">=3.10" requires-python = ">=3.10"

3
requirements.txt
View File

@ -2,4 +2,5 @@ torch
aiia aiia
pillow pillow
torchvision torchvision
sklearn sklearn
https://gitea.fabelous.app/Machine-Learning/AIIA.git

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