Upload folder using huggingface_hub

MLBaseModelDriver.py

@@ -61,7 +61,7 @@ class MLBaseModelDriver:
         model_class = self._import_model_class(model_class_file)
 
         model = model_class(input_dim=4)
-        state_dict = torch.load(model_file, weights_only=False)
+        state_dict = torch.load(model_file, map_location=torch.device("cpu"), weights_only=False)
         model.load_state_dict(state_dict)
         model.eval()