mirror of
https://github.com/nod-ai/AMD-SHARK-Studio.git
synced 2026-02-19 11:56:43 -05:00
773e6ebebf
* Move IREE pins for windows, linux, and pip packaging files. * Remove linux builds from 1.0 nightly workflow It seems that the VMs used for these workflows are no longer available. Removing linux builds since publishing .exe is sufficient for the one-shot nightly workflows we trigger for SHARK-1.0
46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
from setuptools import find_packages
|
|
from setuptools import setup
|
|
|
|
import os
|
|
import glob
|
|
|
|
with open("README.md", "r", encoding="utf-8") as fh:
|
|
long_description = fh.read()
|
|
|
|
PACKAGE_VERSION = os.environ.get("SHARK_PACKAGE_VERSION") or "0.0.5"
|
|
backend_deps = []
|
|
if "NO_BACKEND" in os.environ.keys():
|
|
backend_deps = [
|
|
"iree-compiler==20231212.*",
|
|
"iree-runtime==20231212.*",
|
|
]
|
|
|
|
setup(
|
|
name="nodai-SHARK",
|
|
version=f"{PACKAGE_VERSION}",
|
|
description="SHARK provides a High Performance Machine Learning Framework",
|
|
author="nod.ai",
|
|
author_email="stdin@nod.ai",
|
|
url="https://nod.ai",
|
|
long_description=long_description,
|
|
long_description_content_type="text/markdown",
|
|
project_urls={
|
|
"Code": "https://github.com/nod-ai/SHARK",
|
|
"Bug Tracker": "https://github.com/nod-ai/SHARK/issues",
|
|
},
|
|
classifiers=[
|
|
"Programming Language :: Python :: 3",
|
|
"License :: OSI Approved :: MIT License",
|
|
"Operating System :: OS Independent",
|
|
],
|
|
packages=find_packages(exclude=("examples")),
|
|
python_requires=">=3.9",
|
|
data_files=glob.glob("apps/stable_diffusion/resources/**"),
|
|
install_requires=[
|
|
"numpy",
|
|
"PyYAML",
|
|
"torch-mlir",
|
|
]
|
|
+ backend_deps,
|
|
)
|