Update device_manager.py

device_manager.py

@@ -89,7 +89,11 @@ def device_handler(func):
         processed_kwargs = {k: process_arg(v) for k, v in kwargs.items()}
         
         try:
-            result = await func(*processed_args, **processed_kwargs)
+            # 如果函數是異步的，使用 await
+            if asyncio.iscoroutinefunction(func):
+                result = await func(*processed_args, **processed_kwargs)
+            else:
+                result = func(*processed_args, **processed_kwargs)
             
             # 處理輸出
             if torch.is_tensor(result):