Updated PyTorch to 2.1.2 with CUDA 12.1 index and enabled faiss-gpu. Pinned torch/torchvision/torchaudio versions to prevent mismatches. 9af55c1 verified harshinde commited on 18 days ago
prevent too fast map_to_image calls (returns None) d4b7ad4 verified mikonvergence commited on Mar 5, 2024