search

NJU-PCALab / AddSR

Apache License 2.0
82 stars 2 forks source link

Right way to setup pretrained weights? #17

Closed ppriyank closed 3 months ago

ppriyank commented 3 months ago

pipelines/pipeline_addsr.py test_addsr.py

This weight "/home/amax/workspace/ruixie/SeeSR/preset/ram_swin_large_14m.pth" I downloaded it from the website directly, but it's not mentioned in the Readme: https://huggingface.co/spaces/xinyu1205/recognize-anything/blob/main/ram_swin_large_14m.pth

was it supposed to be set differently?

Additionally, (ram/models/utils.py)

tokenizer = BertTokenizer.from_pretrained('/home/amax/workspace/ruixie/SeeSR/ram/bert')

what would be the right path for this

CSRuiXie commented 3 months ago

We have provided the download link for the 'ram_swin_large_14m.pth' file here in the Readme. It is the same as the one you have downloaded. Additionally, you can download the BERT weights from here.

ppriyank commented 3 months ago

gotcha, thanks