Spaces:

muryshev
/

nn-search-transmap

Sleeping

muryshev commited on Jan 13

Commit

575d4e0

1 Parent(s): a49da61

update

Files changed (1) hide show

business_transaction_map/components/faiss_vector_database.py CHANGED Viewed

@@ -197,7 +197,7 @@ class FaissVectorDatabase:
         if query_dim != index_dim:
             if query_dim < index_dim:
                 # Padding до нужной размерности
-                padded_query = torch.zeros((1, index_dim), dtype=emb_query.dtype)
                 padded_query[:, :query_dim] = emb_query
                 emb_query = padded_query
             else:

         if query_dim != index_dim:
             if query_dim < index_dim:
                 # Padding до нужной размерности
+                padded_query = torch.zeros((1, index_dim), dtype=torch.float32)
                 padded_query[:, :query_dim] = emb_query
                 emb_query = padded_query
             else: