Fix linear 2 substitution index

Signed-off-by: Alex-Brooks <Alex.Brooks@ibm.com>
This commit is contained in:
Alex-Brooks 2025-01-27 12:09:04 -07:00
parent 7905f9dd40
commit 987f76840a

View file

@ -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: