# load embeddingstext_embeddings = load('data/text_embeddings.pt', weights_only=True)image_embeddings = load('data/image_embeddings.pt', weights_only=True)print(text_embeddings.shape)print(image_embeddings.shape)# >> torch.Size([86, 512])# >> torch.Size([17, 512])