search

PRIS-CV / DemoFusion

Let us democratise high-resolution generation! (CVPR 2024)
https://ruoyidu.github.io/demofusion/demofusion.html
1.98k stars 229 forks source link

fix error when there is no `image_lr` but lowres mode tries to move i… #42

Closed jonathandinu closed 8 months ago

jonathandinu commented 8 months ago

…t to cpu.

If lowres=True but image_lr is None the pipeline errors when it tries to move it to the cpu.