Machine-Learning
/
AIIA
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

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

Reviewed-on: #13
This commit is contained in:
Falko Victor Habel 2025-03-01 21:54:36 +00:00
parent 5205a73811 06a2e012e8
commit 8d05481498
5 changed files with 8 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

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

2
pyproject.toml
View File

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

2
requirements.txt
View File

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

6
run.py
View File

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

10
setup.cfg
View File

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