search

Algolzw / daclip-uir

[ICLR 2024] Controlling Vision-Language Models for Universal Image Restoration. 5th place in the NTIRE 2024 Restore Any Image Model in the Wild Challenge.
https://algolzw.github.io/daclip-uir
MIT License
582 stars 30 forks source link

Fix cpu #22

Open edumotya opened 5 months ago

edumotya commented 5 months ago

Fix https://github.com/Algolzw/daclip-uir/issues/21 by simply eliminating the explicit conversion of zero_modules to cuda.

As far as I see create_model will take care of moving everything to cuda https://github.com/Algolzw/daclip-uir/blob/a4e6720fcbf27c3e99840990bea1e9fbb67710dc/da-clip/src/open_clip/factory.py#L219

LukaGiorgadze commented 5 months ago

Thank you, tiny but helpful PR. Did you test on cpu?

edumotya commented 5 months ago

Sure! it works