procgne/Plonk

1from setuptools import setup, find_packages2 3setup(4    name="diff_plonk",5    version="0.1",6    packages=find_packages(),7    install_requires=[8        "torch",9        "torchvision",10        "joblib",11        "wandb",12        "hydra-core",13        "numpy",14        "scipy==1.13.1",15        "pandas",16        "scikit-learn",17        "pytorch-lightning",18        "transformers",19        "accelerate",20        "peft",21        "geos",22        "reverse_geocoder",23        "matplotlib",24        "geoopt",25        "einops",26        "torchdiffeq",27        "webdataset==0.2.57",28        "pytest",29        "streamlit",30        "streamlit-extras",31        "plotly",32    ],33)34