omniparser v2 hf download commands
This commit is contained in:
@@ -56,8 +56,12 @@ There are three components:
|
||||
|
||||
g. Continue from here if you already had the conda environment.
|
||||
|
||||
h. Ensure you have the weights downloaded in weights folder. If not download them with:
|
||||
`for folder in icon_caption_florence icon_detect icon_detect_v1_5; do huggingface-cli download microsoft/OmniParser --local-dir weights/ --repo-type model --include "$folder/*"; done`
|
||||
h. Ensure you have the V2 weights downloaded in weights folder (**ensure caption weights folder is called icon_caption_florence**). If not download them with:
|
||||
```
|
||||
rm -rf weights/icon_detect weights/icon_caption weights/icon_caption_florence
|
||||
for f in icon_detect/{train_args.yaml,model.pt,model.yaml} icon_caption/{config.json,generation_config.json,model.safetensors}; do huggingface-cli download microsoft/OmniParser-v2.0 "$f" --local-dir weights; done
|
||||
mv weights/icon_caption weights/icon_caption_florence
|
||||
```
|
||||
|
||||
h. Navigate to the server directory with `cd OmniParser/omnitool/omniparserserver`
|
||||
|
||||
|
||||
Reference in New Issue
Block a user