va1shn9v / PromptIR

PromptIR: Prompting for All-in-One Blind Image Restoration [NeurIPS 2023]
https://arxiv.org/abs/2306.13090
Other
297 stars 22 forks source link

Cuda out of memory error? #2

Closed zelenooki87 closed 1 year ago

zelenooki87 commented 1 year ago

Is there any chance to inference with tile option to avoid oom errors? tnx

va1shn9v commented 1 year ago

Hello, Thanks for taking an interest in our work, we will be adding tiling inference in the next few days

heorhiikalaichev commented 1 year ago

Faced with the same issue.

Thank you in advance)

va1shn9v commented 1 year ago

Updated, demo.py with tiling options, kindly let me know if this helps