Update model_handler.py

model_handler.py

@@ -22,19 +22,24 @@ import torch
 if not hasattr(torch, "xpu"):
     class _DummyXPU:
         @staticmethod
-        def empty_cache() -> None:
-            # No‑op: nothing to clear on CPU‑only builds.
+        def empty_cache():
             return None
         @staticmethod
-        def is_available() -> bool:
-            # Always report unavailable on CPU‑only builds.
+        def manual_seed(_seed: int):
+            return None
+        @staticmethod
+        def is_available():
             return False
         @staticmethod
-        def device_count() -> int:
-            # Zero XPU devices available.
+        def device_count():
+            return 0
+        @staticmethod
+        def current_device():
             return 0
-    # Attach dummy XPU to torch
-    torch.xpu = _DummyXPU()  # type: ignore[attr-defined]
+        @staticmethod
+        def set_device(_idx: int):
+            return None
+    torch.xpu = _DummyXPU()  # type: ignore
 import time
 from typing import Optional, Tuple, Any
 from transformers import UMT5EncoderModel