Remix.run Logo
hliyan 3 days ago

Wait, your title says this "runs locally"?

ArnavAgrawal03 2 days ago | parent [-]

Yes! 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.