diff --git a/.gitignore b/.gitignore index c064d35..6586727 100644 --- a/.gitignore +++ b/.gitignore @@ -41,3 +41,6 @@ assets/* # Don't ignore .gitignore !.gitignore + +# Allow PyPI setup script +!setup.py diff --git a/requirements.txt b/requirements.txt index 46df8b2..9d73c64 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ psutil~=6.0.0 requests~=2.32.3 numpy<2.0.0 -torch~=1.13.1 +torch~=2.2.0 sentencepiece~=0.2.0 PyYAML~=6.0.2 pynvml~=11.5.3 diff --git a/setup.py b/setup.py new file mode 100644 index 0000000..4104199 --- /dev/null +++ b/setup.py @@ -0,0 +1,21 @@ +from setuptools import setup + +with open('requirements.txt') as f: + required = f.read().splitlines() + +setup( + name='AutoGGUF', + version='v1.7.1', + packages=[''], + url='https://github.com/leafspark/AutoGGUF', + license='apache-2.0', + author='leafspark', + author_email='', + description='automatically quant GGUF models', + install_requires=required, + entry_points={ + 'console_scripts': [ + 'autogguf-gui = src.main' + ] + } +)