update utils

Utils/Utils.py

@@ -597,13 +597,13 @@ def group_themes(tensor, vocabulary, num_groups, groupedKeywordsPath=''):
     # Create the grouped tensor
     num_rows = tensor.size(0)
     num_themes = len(grouped_themes)
-    grouped_tensor = torch.zeros((num_rows, num_themes), dtype=torch.float32)
+    grouped_tensor = torch.zeros((num_rows, num_themes), dtype=torch.float32, device=tensor.device)
 
     # For each word in the vocabulary, add its count to the corresponding theme
     for word_idx, word in enumerate(vocabulary):
         if word in word_to_theme:
             theme_idx = word_to_theme[word]
-            grouped_tensor[:, theme_idx] += tensor[:, word_idx]
+            grouped_tensor[:, theme_idx] += tensor[:, word_idx].to(grouped_tensor.device) 
 
     themes = list(grouped_themes.keys())