Added patch to run models on CPU

nygaardcodecommentclassification/api/models.py

@@ -22,6 +22,16 @@ import torch
 
 from nygaardcodecommentclassification import config
 
+# Patch torch.load to use CPU mapping by default if CUDA is not available
+# This prevents "Attempting to deserialize object on a CUDA device" errors
+_original_torch_load = torch.load
+def _patched_torch_load(f, map_location=None, *args, **kwargs):
+    """Wrapper around torch.load that uses CPU mapping if CUDA unavailable."""
+    if map_location is None and not torch.cuda.is_available():
+        map_location = torch.device('cpu')
+    return _original_torch_load(f, map_location=map_location, *args, **kwargs)
+torch.load = _patched_torch_load
+
 # Configure module logger with explicit handler to ensure visibility
 logger = logging.getLogger("nygaard.models")
 logger.setLevel(logging.DEBUG)