Skip to content

Commit

Permalink
Updates to make torchcfm pip installable
Browse files Browse the repository at this point in the history
  • Loading branch information
atong01 committed Nov 1, 2023
1 parent 734a711 commit bc91ce1
Show file tree
Hide file tree
Showing 3 changed files with 26 additions and 4 deletions.
2 changes: 2 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -171,3 +171,5 @@ slurm*.out
*.jpg

notebooks/figures/

.DS_Store
3 changes: 1 addition & 2 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,8 @@ scipy
scikit-learn
scprep
scanpy
timm
torchdyn>=1.0.7 # 1.0.4 is broken on pypi
pot
torchdiffeq==0.2.3
torchdiffeq
absl-py
clean-fid
25 changes: 23 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,15 +4,36 @@

from setuptools import find_packages, setup

install_requires = [
"torch>=1.11.0",
"torchvision>=0.11.0",
"lightning-bolts",
"matplotlib",
"numpy",
"scipy",
"scikit-learn",
"scprep",
"scanpy",
"torchdyn",
"pot",
"torchdiffeq",
"absl-py",
"clean-fid",
]

version_py = os.path.join(os.path.dirname(__file__), "torchcfm", "version.py")
version = open(version_py).read().strip().split("=")[-1].replace('"', "").strip()
readme = open("README.md").read()
setup(
name="torchcfm",
version=version,
description="Conditional Flow Matching for Fast Continuous Normalizing Flow Training.",
author="Alexander Tong",
author="Alexander Tong, Kilian Fatras",
author_email="[email protected]",
url="https://github.com/atong01/conditional-flow-matching",
install_requires=["torch", "pot", "numpy", "torchdyn"],
install_requires=install_requires,
license="MIT",
long_description=readme,
long_description_content_type="text/markdown",
packages=find_packages(),
)

0 comments on commit bc91ce1

Please sign in to comment.