"slow_conv2d_cpu" not implemented for 'Half'

Viewed 47

I am trying to run some software I found called CLIP Interrogator in CPU mode, as I'd like to preserver VRAM for other operations, and if i run on CUDA I get out of memory errors due to my main program holding ontop VRAM for it's diffusion.

I specifically get the error:

"slow_conv2d_cpu" not implemented for 'Half' from /usr/local/lib/python3.7/dist-packages/torch/nn/modules/conv.py on the function:

    def _conv_forward(self, input: Tensor, weight: Tensor, bias: Optional[Tensor]):
        if self.padding_mode != 'zeros':
            return F.conv2d(F.pad(input, self._reversed_padding_repeated_twice, mode=self.padding_mode),
                            weight, bias, self.stride,
                            _pair(0), self.dilation, self.groups)
        return F.conv2d(input, weight, bias, self.stride,
                        self.padding, self.dilation, self.groups) #<-- Error on this line

Is there a way to force this to full float, or any other work around?

0 Answers
Related