From 58e4fadfc0f09fd335101dc48d53953838efac7c Mon Sep 17 00:00:00 2001
From: Jake Poznanski <jakep@allenai.org>
Date: Fri, 27 Jun 2025 16:16:19 +0000
Subject: [PATCH] torchvision requirement

---
 gantry-train-requirements.txt | 1 +
 1 file changed, 1 insertion(+)

diff --git a/gantry-train-requirements.txt b/gantry-train-requirements.txt
index 441796f..e65d726 100644
--- a/gantry-train-requirements.txt
+++ b/gantry-train-requirements.txt
@@ -1,3 +1,4 @@
+torchvision
 accelerate
 peft
 wandb