From 9c19e150887cb65d0f5ffa5c673c52bc75c6d6d8 Mon Sep 17 00:00:00 2001
From: Joseph Omar <j.omar@soton.ac.uk>
Date: Mon, 2 Dec 2024 15:38:52 +0000
Subject: [PATCH] fix reqs

---
 requirements-extras.txt | 3 +++
 requirements.txt        | 4 +---
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/requirements-extras.txt b/requirements-extras.txt
index ac75fb3..35aecab 100644
--- a/requirements-extras.txt
+++ b/requirements-extras.txt
@@ -1,2 +1,5 @@
 mmcv-full==1.5.0
 mmsegmentation==0.27.0
+submitit
+--extra-index-url https://pypi.nvidia.com
+cuml-cu11
diff --git a/requirements.txt b/requirements.txt
index 04c159c..69a4213 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,6 +6,4 @@ torchmetrics==0.10.3
 fvcore
 iopath
 xformers==0.0.18
-submitit
---extra-index-url https://pypi.nvidia.com
-cuml-cu11
+
-- 
GitLab