From 7966150d90defcbedbe5fc7f207dd51f2a5c05bd Mon Sep 17 00:00:00 2001
From: BuildTools <chenxuzhang45@gmail.com>
Date: Sat, 17 Aug 2024 13:41:21 -0700
Subject: [PATCH] ci: add PyPI setup script

---
 .gitignore       |  3 +++
 requirements.txt |  2 +-
 setup.py         | 21 +++++++++++++++++++++
 3 files changed, 25 insertions(+), 1 deletion(-)
 create mode 100644 setup.py

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'
+        ]
+    }
+)