Пакету Flash Attention (flash_attn) не удается создать колесо в Google Colab из-за выхода из подпроцесса с ошибкой ⇐ Python

[Anonymous]

Я пытаюсь запустить модель Florence-2 в Google Colab, для чего требуется пакет flash_attn. Раньше модель и пакет устанавливались в Colab без проблем, но теперь при попытке установить flash_attn я сталкиваюсь со следующей ошибкой

Код: Выделить всё

Building wheels for collected packages: flash_attn
error: subprocess-exited-with-error

× python setup.py bdist_wheel did not run successfully.
│ exit code: 1
╰─> See above for output.

note: This error originates from a subprocess, and is likely not a problem with pip.
Building wheel for flash_attn (setup.py) ... error
ERROR: Failed building wheel for flash_attn
Running setup.py clean for flash_attn
Failed to build flash_attn
ERROR: Failed to build installable wheels for some pyproject.toml based projects (flash_attn).

Я уже пробовал следующее, но не получилось:

• Обновление точки, колеса и инструменты настройки

Код: Выделить всё

!pip install --upgrade pip wheel setuptools

2.Создаем виртуальную среду в Colab и устанавливаем туда flash_attn
3.Ищем совместимые версии flash_attn и ее зависимостей, но не нашли Мне не удалось выявить какие-либо конфликты версий.
Поскольку flash_attn является критической зависимостью, я подозреваю, что может возникнуть проблема совместимости версий с недавними обновлениями Colab или базовыми библиотеками, но я не могу найти ни одного подробные требования к версии или обходные пути. Кто-нибудь еще сталкивался с этой проблемой или знает ли кто-нибудь совместимые версии или другие решения?
Спасибо за любую помощь в решении этой проблемы с установкой!

Подробнее здесь: https://stackoverflow.com/questions/791 ... lab-due-to