Spaces:
Running
Running
File size: 668 Bytes
c4c7cee |
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 29 30 31 32 33 34 |
from setuptools import setup, find_packages
setup(
name="diff_plonk",
version="0.1",
packages=find_packages(),
install_requires=[
"torch",
"torchvision",
"joblib",
"wandb",
"hydra-core",
"numpy",
"scipy==1.13.1",
"pandas",
"scikit-learn",
"pytorch-lightning",
"transformers",
"accelerate",
"peft",
"geos",
"reverse_geocoder",
"matplotlib",
"geoopt",
"einops",
"torchdiffeq",
"webdataset==0.2.57",
"pytest",
"streamlit",
"streamlit-extras",
"plotly",
],
)
|