You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

38 lines
1.1 KiB

from setuptools import setup, find_packages
from python_coreml_stable_diffusion._version import __version__
with open('README.md') as f:
readme = f.read()
setup(
name='python_coreml_stable_diffusion',
version=__version__,
url='https://github.com/apple/ml-stable-diffusion',
description="Run Stable Diffusion on Apple Silicon with Core ML (Python and Swift)",
long_description=readme,
long_description_content_type='text/markdown',
author='Apple Inc.',
install_requires=[
"coremltools>=6.1",
"diffusers[torch]",
"torch",
"transformers",
"scipy",
"numpy<1.24",
],
packages=find_packages(),
classifiers=[
"Development Status :: 4 - Beta",
"Intended Audience :: Developers",
"Operating System :: MacOS :: MacOS X",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.7",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Topic :: Artificial Intelligence",
"Topic :: Scientific/Engineering",
"Topic :: Software Development",
],
)