From 461d4ff4203749b4c2fce5eda0f3f895f8a25046 Mon Sep 17 00:00:00 2001
From: camenduru <54370274+camenduru@users.noreply.github.com>
Date: Thu, 8 Aug 2024 12:52:42 +0300
Subject: [PATCH] cuda

---
 totoro/model_management.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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()