3 ms·
Wait, your title says this "runs locally"?
by hliyan 2y ago
Wait, your title says this "runs locally"?
- ArnavAgrawal03 2y agoYes! If you're running the local version and it's taking long, that an indication that your GPU isn't being used properly. This can be traced back to the `colpali_embedding_model.py` file, where you can set the device and attention you want PyTorch to use.