Merge pull request 'feat/docs' (#9) from feat/docs into develop

Reviewed-on: #9

.gitea/workflows/embed.yaml

@@ -0,0 +1,37 @@
+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

@@ -0,0 +1,37 @@
+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,17 +1,15 @@
 [build-system]
 requires = ["setuptools>=45", "wheel"]
 build-backend = "setuptools.build_meta"
-
 [project]
 name = "aiunn"
-version = "0.1.2"
+version = "0.1.1"
 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
-sklearn
+scikit-learn
-https://gitea.fabelous.app/Machine-Learning/AIIA.git
+git+https://gitea.fabelous.app/Machine-Learning/AIIA.git

setup.py

@@ -6,9 +6,14 @@ setup(
     packages=find_packages(where="src"),
     package_dir={"": "src"},
     install_requires=[
-        line.strip()
+        "torch",
-        for line in open("requirements.txt")
+        "aiia",
-        if line.strip() and not line.startswith("#")
+        "pillow",
+        "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("hehe")
+    upsampler.save("aiunn")
 
     # Now load using the overridden load method; this will load the complete model.
-    upsampler_loaded = aiuNN.load("hehe", precision="bf16")
+    upsampler_loaded = aiuNN.load("aiunn", precision="bf16")
     print("Updated configuration:", upsampler_loaded.config.__dict__)