third_codes/eland
1
0
Fork 0
mirror of https://github.com/elastic/eland.git synced 2025-07-11 00:02:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Error measuring embedding size for some DPR models (#573)

Browse Source 
Fixes an error unpacking a tuple that contains a single element.
This commit is contained in:
David Kyle 2023-09-19 10:44:15 +01:00 committed by GitHub
parent 5c5ef63a69
commit 301cda8d69
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
eland/ml/pytorch/transformers.py
View File

@ -729,7 +729,7 @@ class TransformerModel:
else: else:
sample_embedding = self._traceable_model.sample_output() sample_embedding = self._traceable_model.sample_output()
if type(sample_embedding) is tuple: if type(sample_embedding) is tuple:
text_embedding, _ = sample_embedding text_embedding = sample_embedding[0]
else: else:
text_embedding = sample_embedding text_embedding = sample_embedding