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
|
|
|
)
|
