特牛网址导航

RuntimeError: “softmax_lastdim_kernel_impl“ not implemented for ‘Half‘_runtimeerror: "softmax_lastdim_kernel_impl" not im-CSDN博客

网友收藏
  • 链接地址:https://blog.csdn.net/lgldl/article/details/123999244
  • 链接标题:RuntimeError: “softmax_lastdim_kernel_impl“ not implemented for ‘Half‘_runtimeerror: "softmax_lastdim_kernel_impl" not im-CSDN博客
  • 所属网站:blog.csdn.net
  • 被收藏次数:6757
文章浏览阅读3k次,点赞3次,收藏3次。将CLIP/clip/model.py第371行函数注释,如下def _convert_weights_to_fp16(l): """ if isinstance(l, (nn.Conv1d, nn.Conv2d, nn.Linear)): l.weight.data = l.weight.data.half() if l.bias is not None: l.bias.data = l.b_runtimeerror: "softmax_lastdim_kernel_impl" not implemented for 'half