Fix linear 2 substitution index
Signed-off-by: Alex-Brooks <Alex.Brooks@ibm.com>
This commit is contained in:
parent
7905f9dd40
commit
987f76840a
1 changed files with 1 additions and 1 deletions
|
@ -46,7 +46,7 @@ def get_tensor_name(name: str) -> str:
|
|||
if "linear_1" in name:
|
||||
name = name.replace("linear_1", "0")
|
||||
if "linear_2" in name:
|
||||
name = name.replace("linear_2", "1")
|
||||
name = name.replace("linear_2", "2")
|
||||
return name
|
||||
|
||||
if "projection" in name:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue