read embeddings to gpu directly.

utils/predict.py

@@ -107,7 +107,7 @@ def xclip_pred(new_desc: dict,
         
         part_embeds = owlvit_processor(text=[ORG_PART_ORDER], return_tensors="pt").to(device)
         if use_precompute_embeddings:
-            image_embeds = torch.load(f'data/image_embeddings/{image_name}.pt').to(device)
+            image_embeds = torch.load(f'data/image_embeddings/{image_name}.pt', weights_only=True, map_location=device).to(device)
         else:
             image_input = owlvit_processor(images=image, return_tensors='pt').to(device)
             image_embeds, _ = model.image_embedder(pixel_values = image_input['pixel_values'])