ddpo-pytorch/setup.py

17 lines
340 B
Python
Raw Normal View History

2023-06-24 04:25:54 +02:00
from setuptools import setup, find_packages
setup(
2023-06-24 09:07:55 +02:00
name="ddpo-pytorch",
version="0.0.1",
2023-06-24 04:25:54 +02:00
packages=["ddpo_pytorch"],
install_requires=[
2023-06-24 09:07:55 +02:00
"ml-collections",
"absl-py",
"diffusers[torch]==0.17.1",
"wandb",
"torchvision",
"inflect==6.0.4",
"transformers==4.30.2",
2023-06-24 04:25:54 +02:00
],
2023-06-24 09:07:55 +02:00
)