CompVis

/

ldm-celebahq-256

like 30

Hello,
After running the code I noticed the error
"
1 # process image
----> 2 image_processed = image.cpu().permute(0, 2, 3, 1)
3 image_processed = (image_processed + 1.0) * 127.5
4 image_processed = image_processed.clamp(0, 255).numpy().astype(np.uint8)

AttributeError: 'DecoderOutput' object has no attribute 'cpu'
"
It seemed to work using
image_processed = image.sample.cpu().permute(0, 2, 3, 1)
I don't really know if this is an issue but it works now