Update model_hf.py

model_hf.py

@@ -431,17 +431,8 @@ class Residual_block(nn.Module):
 
 class Model(PreTrainedModel,nn.Module):
     config_class = SSLConfig
-    def __init__(self,device,config):
+    def __init__(self,config):
         super().__init__(config)
-        # self.model_device = device
-        # print("Attributes and methods in PreTrainedModel:", dir(PreTrainedModel))
-        # if hasattr(PreTrainedModel, "device") and isinstance(getattr(PreTrainedModel, "device"), property):
-        #     print("device is a property in PreTrainedModel")
-        # else:
-        #     print("device is NOT a property in PreTrainedModel")
-
-
-        # print(device)
         # self.model_device ='cuda' if torch.cuda.is_available() else 'cpu'
         # AASIST parameters
         filts = [128, [1, 32], [32, 32], [32, 64], [64, 64]]