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

Not able to test on an image #27

Open Vinayak-VG opened 5 months ago

Vinayak-VG commented 5 months ago

I get this error when I run the code given in the readme file for testing on an image

Traceback (most recent call last): File "run.py", line 6, in model, preprocess = open_clip.create_model_from_pretrained('daclip_ViT-B-32', pretrained=checkpoint) File "/home/vinayak/.local/lib/python3.8/site-packages/open_clip/factory.py", line 437, in create_model_from_pretrained model = create_model( File "/home/vinayak/.local/lib/python3.8/site-packages/open_clip/factory.py", line 215, in create_model raise RuntimeError(f'Model config for {model_name} not found.') RuntimeError: Model config for daclip_ViT-B-32 not found.

Please let us know what has to be done?

Algolzw commented 5 months ago

Hi, which code are you running? I've tried the app.py and it runs very well.

Algolzw commented 5 months ago

Moreover, our project hasn't been merged into the official open_clip repo. Thus you probably should use our open_clip that contains the modified code for our DA-CLIP.

lwtgithublwt commented 3 months ago

RuntimeError: Model config for daclip_ViT-B-32 not found. I'm not sure if you have solved this problem. Can you answer it