trying to fix zero-gpu

app.py

@@ -16,7 +16,7 @@ import torch
 from models.models import BiFormer
 
 
-DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+DEVICE = torch.device('cuda')
 # DEVICE = torch.device('cpu')
 
 # Configure logging

data_loading/feature_extractor.py

@@ -14,7 +14,7 @@ from transformers import (
 from data_loading.pretrained_extractors import EmotionModel, get_model_mamba, Mamba
 
 
-DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+DEVICE = torch.device('cuda')
 # DEVICE = torch.device('cpu')
 
 

data_loading/pretrained_extractors.py

@@ -14,7 +14,7 @@ from einops import rearrange, einsum
 from torch import Tensor
 from einops import rearrange
 
-DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+DEVICE = torch.device('cuda')
 # DEVICE = torch.device('cpu')
 
 ## Audio models