Problem
in only passes a when :
On any non-CUDA accelerator (Ascend NPU, Apple MPS, Intel XPU, AMD ROCm) the Qwen2.5-Omni model silently loads onto the CPU, so the scene metric runs on CPU (slow) even when a GPU/NPU is available. The model weights are loaded via .
Expected
Follow the actual device instead of assuming CUDA: set for every non-CPU device.
Environment
- repo: hao-ai-lab/FastVideo
- device: Ascend 910B (torch.accelerator.current_accelerator() == npu)
Problem
in only passes a when :
On any non-CUDA accelerator (Ascend NPU, Apple MPS, Intel XPU, AMD ROCm) the Qwen2.5-Omni model silently loads onto the CPU, so the scene metric runs on CPU (slow) even when a GPU/NPU is available. The model weights are loaded via .
Expected
Follow the actual device instead of assuming CUDA: set for every non-CPU device.
Environment