🐛 fix hf space error ZeroGPU has not been initialized

lenML · Jun 7, 2024 · 562e17c · 562e17c
1 parent 7585282
commit 562e17c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/devices/devices.py b/modules/devices/devices.py
@@ -111,7 +111,7 @@ def torch_gc():
 
 
 cpu: torch.device = torch.device("cpu")
-device: torch.device = get_optimal_device()
+device: torch.device = None
 dtype: torch.dtype = torch.float32
 dtype_dvae: torch.dtype = torch.float32
 dtype_vocos: torch.dtype = torch.float32