diff --git a/totoro/model_management.py b/totoro/model_management.py
index b039524f6..b5ecc57e1 100644
--- a/totoro/model_management.py
+++ b/totoro/model_management.py
@@ -515,7 +515,7 @@ def unet_offload_device():
     if vram_state == VRAMState.HIGH_VRAM:
         return get_torch_device()
     else:
-        return torch.device("cpu")
+        return torch.device("cuda")
 
 def unet_inital_load_device(parameters, dtype):
     torch_dev = get_torch_device()