DeepSeek-VL icon indicating copy to clipboard operation
DeepSeek-VL copied to clipboard

RuntimeError: "triu_tril_cuda_template" not implemented for 'BFloat16'

Open sulaiman1988 opened this issue 9 months ago • 1 comments

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

sulaiman1988 avatar May 06 '24 13:05 sulaiman1988

pip install transformers==4.36.2

yangtianyu92 avatar May 22 '24 09:05 yangtianyu92