spacey

src/visual_qa.py

@@ -67,7 +67,7 @@ def main_create_image_collection(df_emb, collection_name="clip_image_embeddings"
 
 
 model_name = "patrickjohncyh/fashion-clip"
-device = "mps"
+device = torch.device("mps" if torch.backends.mps.is_available() else "cpu")
 model = CLIPModel.from_pretrained(model_name).to(device)
 processor = CLIPProcessor.from_pretrained(model_name)