From 04d4ffdb9ba77d3c9456eeb1498644c68cca82b7 Mon Sep 17 00:00:00 2001
From: Jozef Fulop <fulop.jozef1@gmail.com>
Date: Wed, 25 Feb 2026 14:28:22 +0100
Subject: [PATCH] fix(generators): handle CPU sentinel in
 SequenceRNN.attachToGPUs

---
 drugex/training/generators/sequence_rnn.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/drugex/training/generators/sequence_rnn.py b/drugex/training/generators/sequence_rnn.py
index 8583d73..558a6c6 100644
--- a/drugex/training/generators/sequence_rnn.py
+++ b/drugex/training/generators/sequence_rnn.py
@@ -43,7 +43,10 @@ def attachToGPUs(self, gpus):
         -------
         None
         """
-        self.device = torch.device(f'cuda:{gpus[0]}')
+        if gpus[0] == -1:
+            self.device = torch.device('cpu')
+        else:
+            self.device = torch.device(f'cuda:{gpus[0]}')
         self.to(self.device)
         self.gpus = (gpus[0],)