fix tested on colab CPU

app.py

@@ -1,31 +1,30 @@
 import sys
 import torch
-import torch._dynamo.device_interface
+# import torch._dynamo.device_interface
+import torch.utils._triton
 
 # Hack unsloth to work without GPU
 def fake_triton_check():
     return False  # Prevents `bitsandbytes` from requiring GPU
 def fake_is_available():
     return True
-def fake_device_capability():
+def fake_device_capability(*args, **kwargs):
     return (8, 0)
 def fake_current_device():
     return 0
-
-class Worker:
-    def __init__(self):
-        pass
-
-    def get_device_properties(self):
-        return FakeDeviceProperties()
-class FakeDeviceProperties:
-    major = 7  # Mimic a compatible GPU
-    minor = 5
+def fake_device_count():
+    return 1
+def has_triton():
+    return False
+def get_fake_stream(*args, **kwargs):
+    return 0
 
 sys.modules["torch"].cuda.is_available = fake_is_available
 sys.modules["torch"].cuda.get_device_capability = fake_device_capability
 sys.modules["torch"].cuda.current_device = fake_current_device
-sys.modules["torch._dynamo.device_interface"].Worker = Worker()
+sys.modules["torch"].cuda.device_count = fake_device_count
+sys.modules["torch.utils._triton"].has_triton = has_triton
+sys.modules["torch._C"]._cuda_getCurrentRawStream = get_fake_stream
 
 import streamlit as st
 from unsloth.chat_templates import get_chat_template, CHAT_TEMPLATES