nodes/kijai-ComfyUI-CCSR
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update tilevae.py

Browse Source
This commit is contained in:
kijai
2024-01-17 23:48:43 +02:00
parent ba94bf899d
commit 795d9c394d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/tilevae.py
+1 -1
View File
@@ -61,7 +61,7 @@ import torch.nn.functional as F
import comfy.model_management import comfy.model_management
cpu = torch.device("cpu") 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 = torch.float16
dtype_vae = torch.float16 dtype_vae = torch.float16
dtype_unet = torch.float16 dtype_unet = torch.float16