pip install FlagEmbedding[finetune] - ModuleNotFoundError: No module named 'torch', but it is

[mbykov]

uv pip install 'FlagEmbedding[finetune]'

Resolved 95 packages in 31ms
× Failed to build flash-attn==2.8.2
├─▶ The build backend returned an error
╰─▶ Call to setuptools.build_meta:__legacy__.build_wheel failed (exit status: 1)
[stderr]
.....
ModuleNotFoundError: No module named 'torch'

but it is:

uv pip list | grep torch
pytorch-triton 3.4.0+git11ec6354
torch 2.9.0.dev20250724+cu128
torchao 0.12.0
torchaudio 2.8.0.dev20250724+cu128
torchdata 0.11.0
torchtune 0.6.1
torchvision 0.24.0.dev20250724+cu128

I have RTX 5060 card, so I installed torch with nightly build:

uv pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128 --upgrade --force-reinstall

and it works ok.

But what could be done now?

[mbykov]

but it's not urgent and not critical. Trainer works ok without flash-attn. I wrote it just in case.