Merge pull request #61 from timothybrooks/patch-1

Add dependencies to setup
richzhang · Aug 25, 2021 · 3dd44f5 · 3dd44f5
2 parents b1995a2 + b2ca903
commit 3dd44f5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -14,6 +14,7 @@
  packages=['lpips'],
  package_data={'lpips': ['weights/v0.0/*.pth','weights/v0.1/*.pth']},
  include_package_data=True,
+ install_requires=["torch>=0.4.0", "torchvision>=0.2.1", "numpy>=1.14.3", "scipy>=1.0.1", "tqdm>=4.28.1"],
  classifiers=[
  "Programming Language :: Python :: 3",
  "License :: OSI Approved :: BSD License",