From 3985d424c090795347bd985db7eac2d8e2510b6f Mon Sep 17 00:00:00 2001
From: Alex Bezdieniezhnykh <zxsanny@gmail.com>
Date: Sun, 6 Apr 2025 18:48:24 +0300
Subject: [PATCH] update requirements.txt

---
 requirements.txt | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index e1c6607..3ff9b32 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,10 @@
 huggingface_hub
 ultralytics
+--extra-index-url https://download.pytorch.org/whl/cu121
+torch==2.3.0+cu121
+torchvision==0.18.0+cu121
+torchaudio==2.3.0+cu121
 albumentations
-
 opencv-python
 matplotlib
 PyYAML