From 78b593c137fc7b2bd40ad7e88959cc2a088b0486 Mon Sep 17 00:00:00 2001
From: David Bau <david.bau@gmail.com>
Date: Wed, 1 Jun 2022 09:53:43 -0400
Subject: [PATCH] Add torchvision dependency.

---
 setup.cfg | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.cfg b/setup.cfg
index 1b17fd2..9a2b7bd 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -22,3 +22,4 @@ python_requires = >=3.7
 install_requires =
     numpy
     torch
+    torchvision