added support for sbert/distilbert model
This commit is contained in:
parent
0867e20bc2
commit
9a7665c455
1 changed files with 0 additions and 1 deletions
|
@ -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:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue