aurelio-ai
/

xclip-base-patch16-zero-shot

Video Classification

feature-extraction

Inference Endpoints

Model card Files Files and versions Community

juanpablomesa commited on Nov 13, 2023

Commit

a0a72c0

·

1 Parent(s): c45d9fe

Removed normalization of embeddings

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -122,8 +122,8 @@ class EndpointHandler:
         batch_emb = batch_emb.cpu().detach().numpy()
         # NORMALIZE
-        self.logger.info("Normalizing numpy array")
-        batch_emb = batch_emb.T / np.linalg.norm(batch_emb, axis=1)
         # transpose back to (21, 512)
         self.logger.info("Converting to list")

         batch_emb = batch_emb.cpu().detach().numpy()
         # NORMALIZE
+        # self.logger.info("Normalizing numpy array")
+        # batch_emb = batch_emb.T / np.linalg.norm(batch_emb, axis=1)
         # transpose back to (21, 512)
         self.logger.info("Converting to list")