1580158
/
automatic
mirror of https://github.com/vladmandic/automatic
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor taesd

pull/1952/head
Vladimir Mandic 2023-08-05 09:38:11 +02:00
parent 095ef45d3f
commit e2ee02a1e8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_samplers_common.py
View File

@ -34,7 +34,7 @@ def single_sample_to_image(sample, approximation=None):
elif approximation == 3:
# x_sample = sample * 1.5
# x_sample = sd_vae_taesd.model()(x_sample.to(devices.device, devices.dtype).unsqueeze(0))[0].detach()
x_sample = sd_vae_taesd.decode(sample, 'sd')
x_sample = sd_vae_taesd.decode(sample)
else:
shared.log.warning(f"Unknown image decode type: {approximation}")
return Image.new(mode="RGB", size=(512, 512))