[fix] base features and search features should be in the same device

algorithm.py

@@ -116,7 +116,7 @@ class ImageSearch():
 
         base_product_features /= base_product_features.norm(dim=-1, keepdim=True)
 
-        wait2search_image_features = torch.from_numpy(self.encode_image(wait2search_image_list))
+        wait2search_image_features = torch.from_numpy(self.encode_image(wait2search_image_list)).to(self.device)
         wait2search_image_features /= wait2search_image_features.norm(dim=-1, keepdim=True)
 
         similarity = wait2search_image_features @ base_product_features.T