mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 01:04:32 +02:00
7d6a98bc62
get_cu_seqlens creates a cu_seqlens tensor with device="cuda", which crashes on non-CUDA devices (e.g. Ascend NPU raises "Torch not compiled with CUDA enabled"). Create it on the same device as the input text_mask instead, which is device-agnostic.