RuntimeError: "triu_tril_cuda_template" not implemented for 'BFloat16'

deepseek-ai / DeepSeek-VL

DeepSeek-VL: Towards Real-World Vision-Language Understanding

MIT License

2.03k stars 190 forks source link

Open sulaiman1988 opened 5 months ago

sulaiman1988 commented 5 months ago

RuntimeError: "triu_tril_cuda_template" not implemented for 'BFloat16' torch 2.0.1 torchaudio 2.0.1 torchvision 0.15.2

yangtianyu92 commented 4 months ago

pip install transformers==4.36.2