-
Notifications
You must be signed in to change notification settings - Fork 15
/
Copy pathsetup.py
28 lines (27 loc) · 806 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
from setuptools import setup, find_packages
setup(
name="attention_map_diffusers",
version="0.1.5",
author="wooyeolbaek",
author_email="[email protected]",
description="attention map for diffusers",
long_description=open("README.md").read(),
long_description_content_type="text/markdown",
url="https://github.com/wooyeolBaek/attention-map-diffusers",
packages=find_packages(),
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
],
python_requires='>=3.7',
install_requires=[
"diffusers>=0.29.0",
"accelerate",
"transformers",
"einops",
"torchvision",
"protobuf",
"sentencepiece",
],
)