Update model_hf.py

model_hf.py

@@ -433,13 +433,13 @@ class Model(PreTrainedModel,nn.Module):
     config_class = SSLConfig
     def __init__(self,config):
         super().__init__(config)
-        # self.model_device ='cuda' if torch.cuda.is_available() else 'cpu'
+        self.model_device ='cuda' if torch.cuda.is_available() else 'cpu'
         # AASIST parameters
         filts = [128, [1, 32], [32, 32], [32, 64], [64, 64]]
         gat_dims = [64, 32]
         pool_ratios = [0.5, 0.5, 0.5, 0.5]
         temperatures =  [2.0, 2.0, 100.0, 100.0]
-        self.model_device = device
+        # self.model_device = device
 
 
         ####