.gitea/workflows/embed.yaml

@@ -1,37 +0,0 @@
-name: Run VectorLoader Script
-
-on:
-  push:
-    branches:
-      - main
-
-jobs:
-  Explore-Gitea-Actions:
-    runs-on: ubuntu-latest
-    container: catthehacker/ubuntu:act-latest
-    steps:
-      - name: Check out repository code
-        uses: actions/checkout@v4
-
-      - name: Set up Python
-        uses: actions/setup-python@v3
-        with:
-          python-version: '3.11.7'
-
-      - name: Clone additional repository
-        run: |
-          git config --global credential.helper cache
-          git clone https://fabel:${{ secrets.CICD }}@gitea.fabelous.app/fabel/VectorLoader.git                              
-
-      - name: Install dependencies
-        run: |
-          cd VectorLoader
-          python -m pip install --upgrade pip
-          pip install -r requirements.txt                              
-
-      - name: Run vectorizing
-        env:
-          VECTORDB_TOKEN: ${{ secrets.VECTORDB_TOKEN }}
-        run: |
-          cd VectorLoader
-          python -m src.run --full                              

.gitea/workflows/test.yaml

@@ -1,37 +0,0 @@
-name: Gitea Actions For AIIA
-run-name: ${{ gitea.actor }} is testing out Gitea Actions 🚀
-on: [push]
-
-jobs:
-  Explore-Gitea-Actions:
-    runs-on: ubuntu-latest
-    container: catthehacker/ubuntu:act-latest
-    steps:
-      - name: Check out repository code
-        uses: actions/checkout@v4
-      
-      - name: Set up Python
-        uses: actions/setup-python@v5
-        with:
-          python-version: '3.11.7'
-      
-      - name: Cache pip and model
-        uses: actions/cache@v3
-        with:
-          path: |
-            ~/.cache/pip
-            ./fabel                                                
-          key: ${{ runner.os }}-pip-model-${{ hashFiles('requirements.txt', 'requirements-dev.txt') }}
-
-          restore-keys: |
-            ${{ runner.os }}-pip-model-                                                
-      
-      - name: Install dependencies
-        run: |
-          python -m pip install --upgrade pip
-          pip install -r requirements.txt
-          pip install -r requirements-dev.txt          
-          pip install -e .                                              
-      - name: Run tests
-        run: |
-          pytest tests/                                

pyproject.toml

@@ -1,15 +1,17 @@
 [build-system]
 requires = ["setuptools>=45", "wheel"]
 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"
 license = {file = "LICENSE"}
 authors = [
-{name = "Falko Habel", email = "falko.habel@gmx.de"},
+    {name = "Falko Habel", email = "falko.habel@gmx.de"},
 ]
+
 [project.urls]
-"Homepage"="https://gitea.fabelous.app/Machine-Learning/aiuNN"
+"Homepage" = "https://gitea.fabelous.app/Machine-Learning/aiuNN"

requirements.txt

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

setup.py

@@ -6,14 +6,9 @@ setup(
     packages=find_packages(where="src"),
     package_dir={"": "src"},
     install_requires=[
-        "torch",
+        line.strip()
-        "aiia",
+        for line in open("requirements.txt")
-        "pillow",
+        if line.strip() and not line.startswith("#")
-        "torchvision",
-        "scikit-learn",
-    ],
-    dependency_links=[
-        "git+https://gitea.fabelous.app/Machine-Learning/AIIA.git#egg=aiia"
     ],
     python_requires=">=3.10",
 )

src/aiunn/upsampler/aiunn.py

@@ -149,8 +149,8 @@ if __name__ == "__main__":
     upsampler = aiuNN(base_model, config=ai_config)
 
     # Save the model (both configuration and weights).
-    upsampler.save("aiunn")
+    upsampler.save("hehe")
 
     # Now load using the overridden load method; this will load the complete model.
-    upsampler_loaded = aiuNN.load("aiunn", precision="bf16")
+    upsampler_loaded = aiuNN.load("hehe", precision="bf16")
     print("Updated configuration:", upsampler_loaded.config.__dict__)