included requirement for aiia

pyproject.toml

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

requirements.txt

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

setup.py

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

src/aiunn/__init__.py

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

src/aiunn/upsampler/aiunn.py

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