From 4e00e8a071e61442ec9711aec7f2112cf4a33e40 Mon Sep 17 00:00:00 2001
From: ChrisGadek1 <krzysztof.gadek15@gmail.com>
Date: Tue, 23 Jun 2026 22:49:04 +0200
Subject: [PATCH] Replace "pytorch-gpu" with "torch" in dependencies

---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 80b81eb..e4b748a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -31,7 +31,7 @@ dependencies = [
     "mamba",
     "transformers",
     "tqdm",
-    "pytorch-gpu",
+    "torch",
     "esm"
 ]