Patryk Wychowaniec 7157e5d064 interrogate: Fix CLIP-interrogation on CPU ...

Currently, trying to perform CLIP interrogation on a CPU fails, saying:

```
RuntimeError: "slow_conv2d_cpu" not implemented for 'Half'
```

This merge request fixes this issue by detecting whether the target
device is CPU and, if so, force-enabling `--no-half` and passing
`device="cpu"` to `clip.load()` (which then does some extra tricks to
ensure it works correctly on CPU).

2022-10-21 09:52:12 +03:00

6.6 KiB

Raw Blame History

View Raw