Merge pull request 'feat/fix_saving' (#13) from feat/fix_saving into main

Reviewed-on: #13

README.md

@@ -25,7 +25,7 @@ That's it! You should now have the AIIA package installed and ready to use.
 ## Example Usage: 
 ```Python
 from aiia.model import AIIABase
-from aiia.model.config import AIIAConfig
+from aiia.model import AIIAConfig
 from aiia.pretrain import Pretrainer
 
 # Create your model

pyproject.toml

@@ -24,7 +24,7 @@ dependencies = [
     "pytest",
     "pillow"
 ]
-requires-python = ">=3.7"
+requires-python = ">=3.10"
 classifiers = [
     "Programming Language :: Python :: 3",
     "License :: OSI Approved :: CC BY-NC 4.0 License",

requirements.txt

@@ -1,4 +1,4 @@
-torch>=4.5.0
+torch>=2.5.0
 numpy
 tqdm
 pytest

run.py

@@ -1,6 +0,0 @@
-
-
-
-from aiia import AIIABase
-
-model = AIIABase.load(path="AIIA-base-512", precision="bf16")

setup.cfg

@@ -1,7 +1,7 @@
 [metadata]
 name = aiia
-version = 0.1.0
-author = Your Name
+version = 0.1.2
+author = Falko Habel
 author_email = falko.habel@gmx.de
 description = AIIA deep learning model implementation
 long_description = file: README.md
@@ -9,16 +9,16 @@ long_description_content_type = text/markdown
 url = https://gitea.fabelous.app/Maschine-Learning/AIIA.git
 classifiers =
     Programming Language :: Python :: 3
-    License :: OSI Approved :: MIT License
+    License :: OSI Approved :: CC BY-NC 4.0 License
     Operating System :: OS Independent
 
 [options]
 package_dir =
     = src
 packages = find:
-python_requires = >=3.7
+python_requires = >=3.10
 install_requires =
-    torch>=1.8.0
+    torch>=2.5.0
     numpy>=1.19.0
     tqdm>=4.62.0