Typo and missing package in the requirements

[yike5460]

Please consider to fix the typo and missing package (flash_attn) in the requirement.txt

The correct should be

einops==0.8.0
flash-attn==2.6.3

Also please consider to give the sample command to download the weight in HF, mine for your reference:

Install HF CLI if not installed:

pip install -U "huggingface_hub[cli]"

Download the weight from the entire repo:

huggingface-cli download microsoft/OmniParser

Or specific weight:

cd weights
huggingface-cli download microsoft/OmniParser icon_detect/model.safetensors --local-dir .
huggingface-cli download microsoft/OmniParser icon_detect/model.yaml --local-dir .
huggingface-cli download microsoft/OmniParser icon_caption_florence/config.json --local-dir .
huggingface-cli download microsoft/OmniParser icon_caption_florence/generation_config.json --local-dir .
huggingface-cli download microsoft/OmniParser icon_caption_florence/model.safetensors --local-dir .
huggingface-cli download microsoft/OmniParser icon_caption_blip2/config.json --local-dir .
huggingface-cli download microsoft/OmniParser icon_caption_blip2/generation_config.json --local-dir .
huggingface-cli download microsoft/OmniParser icon_caption_blip2/pytorch_model-00001-of-00002.bin --local-dir .
huggingface-cli download microsoft/OmniParser icon_caption_blip2/pytorch_model-00002-of-00002.bin --local-dir .
huggingface-cli download microsoft/OmniParser icon_caption_blip2/pytorch_model-00001-of-00002.safetensors --local-dir .
huggingface-cli download microsoft/OmniParser icon_caption_blip2/pytorch_model-00002-of-00002.safetensors --local-dir .
huggingface-cli download microsoft/OmniParser icon_caption_blip2/pytorch_model.bin.index.json --local-dir .

[Nempickaxe]

@yike5460 tried installing flash-attn with T4 GPU on colab, the notebook crashed... Is there a possiblity of having some work around for this?

[abrichr]

@Nempickaxe see https://github.com/microsoft/OmniParser/pull/52 for automated downloading using huggingface_hub in Dockerfile and automatic deployment to EC2 in deploy.py.