LLM/airllm
1
0
Fork 0
mirror of https://github.com/0xSojalSec/airllm.git synced 2026-05-01 11:26:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' of github.com:lyogavin/airllm

Browse Source
This commit is contained in:
Yu Li
2024-09-20 22:19:40 -05:00
parent 40c15e61e2 6195bb4fc1
commit b8c69d7175
2 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
air_llm/setup.py
View File

@@ -1,3 +1,4 @@
import sys
import setuptools
from setuptools.command.install import install
import subprocess
@@ -6,7 +7,10 @@ import subprocess
class PostInstallCommand(install):
def run(self):
install.run(self)
subprocess.check_call(["pip", "install", "--upgrade", "transformers"])
try:
subprocess.check_call([sys.executable, "-m", "pip", "install", "--upgrade", "transformers"])
except subprocess.CalledProcessError:
print("Warning: Unable to upgrade transformers package. Please upgrade manually.")
# Windows uses a different default encoding (use a consistent encoding)
with open("README.md", "r", encoding="utf-8") as fh: