diff --git a/README.md b/README.md index a3091ec..333838b 100644 --- a/README.md +++ b/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 diff --git a/pyproject.toml b/pyproject.toml index 41f1686..baec4b1 100644 --- a/pyproject.toml +++ b/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", diff --git a/requirements.txt b/requirements.txt index 06e8438..8db58b1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -torch>=4.5.0 +torch>=2.5.0 numpy tqdm pytest diff --git a/run.py b/run.py deleted file mode 100644 index f9fa8a9..0000000 --- a/run.py +++ /dev/null @@ -1,6 +0,0 @@ - - - -from aiia import AIIABase - -model = AIIABase.load(path="AIIA-base-512", precision="bf16") \ No newline at end of file diff --git a/setup.cfg b/setup.cfg index fb45363..ff97663 100644 --- a/setup.cfg +++ b/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