Fixed a bug in embedding_result

This commit is contained in:
kinaryml 2022-11-12 08:55:56 -08:00
parent a3788a23bc
commit a7ed160a8e

View File

@ -51,7 +51,7 @@ class Embedding:
bytearray(pb2_obj.quantized_embedding.values))
float_embedding = np.array(pb2_obj.float_embedding.values, dtype=float)
if not quantized_embedding:
if not pb2_obj.quantized_embedding.values:
return Embedding(
embedding=float_embedding,
head_index=pb2_obj.head_index,