Update utils/system/memory_manager.py

utils/system/memory_manager.py

@@ -20,7 +20,9 @@ class MemoryManager:
     Comprehensive memory management for video processing applications
     """
     
-    def __init__(self, device: torch.device, memory_limit_gb: Optional[float] = None):
+    def __init__(self, device, memory_limit_gb: Optional[float] = None):
+        if isinstance(device, str):
+            device = torch.device(device)
         self.device = device
         self.gpu_available = device.type in ['cuda', 'mps']
         self.memory_limit_gb = memory_limit_gb