deepseek-ai / deepseek-ai/DeepSeek-VL2
Does not work on CPU
- Dominant language
- Python
- Stars
- 5.4k
- Forks
- 1.8k
- PR merge metrics
- No merged PRs in 30d
Description
I tried running DeepSeek-vl2-tiny on the CPU. But it throws an error.
Does it support CPU inference?
I tried backtracking, and an error is being thrown from the memory_efficient_attention of the "formers" package. When I checked the operator bindings, I saw that the operator class IMPL is only registered for the CUDA dispatch key.
So, how can I run this model on a CPU?
Below is the output:
NotImplementedError: No operator found for `memory_efficient_attention_forward` with inputs:
query : shape=(19, 729, 16, 72) (torch.bfloat16)
key : shape=(19, 729, 16, 72) (torch.bfloat16)
value : shape=(19, 729, 16, 72) (torch.bfloat16)
attn_bias :
p : 0.0
`ckF` is not supported because:
device=cpu (supported: {'cuda'})
bf16 is only supported on A100+ GPUs
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.