YogeLiu/TexTeller
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove onnxruntime-gpu

Browse Source
This commit is contained in:
三洋三洋
2024-06-23 22:13:51 +08:00
parent 6793142557
commit 838febf48c
1 changed files with 4 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
setup.py
View File

@@ -1,11 +1,10 @@
from setuptools import setup, find_packages from setuptools import setup, find_packages
import platform
# Define the base dependencies # Define the base dependencies
install_requires = [ install_requires = [
"torch", "torch",
"torchvision", "torchvision",
"torchaudio",
"transformers", "transformers",
"datasets", "datasets",
"evaluate", "evaluate",
@@ -20,18 +19,13 @@ install_requires = [
"streamlit-paste-button", "streamlit-paste-button",
"shapely", "shapely",
"pyclipper", "pyclipper",
"optimum[exporters]"
]
# Add platform-specific dependencies "optimum[exporters]",
if platform.system() == "Darwin": ]
install_requires.append("onnxruntime")
else:
install_requires.append("onnxruntime-gpu")
setup( setup(
name="texteller", name="texteller",
version="0.1.0", version="0.1.2",
author="OleehyO", author="OleehyO",
author_email="1258009915@qq.com", author_email="1258009915@qq.com",
description="A meta-package for installing dependencies", description="A meta-package for installing dependencies",