Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

improve live preview

Browse Source
This commit is contained in:
Vladimir Mandic
2024-01-29 08:53:48 -05:00
parent 753bfebe1c
commit 682da607a9
3 changed files with 26 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_vae_approx.py
+3 -3
View File
@@ -43,7 +43,7 @@ def nn_approximation(sample): # Approximate NN
sd_vae_approx_model.load_state_dict(approx_weights)
sd_vae_approx_model.eval()
sd_vae_approx_model.to(devices.device, sample.dtype)
shared.log.debug(f'Load VAE decode approximate: model="{model_path}"')
shared.log.debug(f'VAE load: type=approximate model={model_path}')
try:
in_sample = sample.to(devices.device).unsqueeze(0)
sd_vae_approx_model.to(devices.device, devices.dtype)
@@ -51,7 +51,7 @@ def nn_approximation(sample): # Approximate NN
x_sample = x_sample[0].detach().cpu()
return x_sample
except Exception as e:
shared.log.error(f'Decode approximate: {e}')
shared.log.error(f'VAE decode approximate: {e}')
return sample
@@ -77,5 +77,5 @@ def cheap_approximation(sample): # Approximate simple
x_sample = nn.functional.conv2d(sample, weights, bias) # pylint: disable=not-callable
return x_sample
except Exception as e:
shared.log.error(f'Decode simple: {e}')
shared.log.error(f'VAE decode simple: {e}')
return sample