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

Default to fp16 unet with 'auto'

Browse Source
This commit is contained in:
Kijai
2024-03-19 16:25:54 +02:00
parent dbc22b88fb
commit 14480f926a
2 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+4 -4
View File
@@ -186,14 +186,14 @@ class SUPIR_Upscale:
if diffusion_dtype == 'auto':
try:
if mm.should_use_fp16():
print("Diffusion using fp16")
dtype = torch.float16
model_dtype = 'fp16'
if mm.should_use_bf16():
print("Diffusion using bf16")
dtype = torch.bfloat16
model_dtype = 'bf16'
elif mm.should_use_fp16():
print("Diffusion using using fp16")
dtype = torch.float16
model_dtype = 'fp16'
else:
print("Diffusion using using fp32")
dtype = torch.float32
nodes_v2.py
+6 -5
View File
@@ -259,6 +259,7 @@ class SUPIR_first_stage:
mm.unload_all_models()
if encoder_dtype == 'auto':
try:
if mm.should_use_bf16():
print("Encoder using bf16")
vae_dtype = 'bf16'
@@ -610,14 +611,14 @@ class SUPIR_model_loader:
if diffusion_dtype == 'auto':
try:
if mm.should_use_bf16():
if mm.should_use_fp16():
print("Diffusion using fp16")
dtype = torch.float16
model_dtype = 'fp16'
elif mm.should_use_bf16():
print("Diffusion using bf16")
dtype = torch.bfloat16
model_dtype = 'bf16'
elif mm.should_use_fp16():
print("Diffusion using using fp16")
dtype = torch.float16
model_dtype = 'fp16'
else:
print("Diffusion using using fp32")
dtype = torch.float32