Test output is black
I want to test StableSR using your test_example, the result is a black image.
I download pretrained_SDmodel v2-1_512-ema-pruned.ckpt ,pretrained_CFWmodel vqgan_cfw_00011.ckpt and stablesr_000117.ckpt.
And I run the code python scripts/sr_val_ddpm_text_T_vqganfin_oldcanvas_tile.py --config configs/stableSRNew/v2-finetune_text_T_512.yaml --ckpt pretrained_turbo/stablesr_000117.ckpt --init-img inputs/test_example --outdir OUT_PATH --ddpm_steps 4 --dec_w 0.5 --seed 42 --n_samples 1 --vqgan_ckpt pretrained_CFWmodel/vqgan_cfw_00011.ckpt --colorfix_type wavelet --upscale 1
The input is 720720
The output is also 720720 but black
I don't know which part went wrong.
I changed ckpt_path to pretrained_SDmodel/v2-1_512-ema-pruned.ckpt in v2-finetune_text_T_512.yaml
Hi. I did not meet such a problem. You may check if the output is None. If so, you may need to avoid using autocasting for the autoencoder part. This is not very common.