Update tilevae.py

2024-01-17 23:48:43 +02:00
parent ba94bf899d
commit 795d9c394d
1 changed files with 1 additions and 1 deletions
@@ -61,7 +61,7 @@ import torch.nn.functional as F
 import comfy.model_management

 cpu = torch.device("cpu")
-device = device_interrogate = device_gfpgan = device_esrgan = device_codeformer = torch.device("cuda")
+device = device_interrogate = device_gfpgan = device_esrgan = device_codeformer = comfy.model_management.get_torch_device()
 dtype = torch.float16
 dtype_vae = torch.float16
 dtype_unet = torch.float16