added support for sbert/distilbert model

This commit is contained in:
Andrew 2023-11-17 14:01:14 -05:00
parent 0867e20bc2
commit 9a7665c455

View file

@ -254,7 +254,6 @@ class TensorNameMap:
return result[0] return result[0]
def __getitem__(self, key: str) -> str: def __getitem__(self, key: str) -> str:
#print ("getting", key)
try: try:
return self.mapping[key][1] return self.mapping[key][1]
except KeyError: except KeyError: