fix generate script

Asthestarsfalll · May 13, 2024 · 8ad2d94 · 8ad2d94
1 parent 35a4374
commit 8ad2d94
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/frontends/paddle/tests/test_models/gen_scripts/generate_tril_triu.py b/src/frontends/paddle/tests/test_models/gen_scripts/generate_tril_triu.py
@@ -29,7 +29,7 @@ def triu(name: str, x, diagonal=0, dtype="float32"):
         saveModel(
             name,
             exe,
-            feedkeys=["x"],
+            feed_vars=[node_x],
             fetchlist=[triu_outs],
             inputs=[x],
             outputs=outs,
@@ -54,7 +54,7 @@ def tril(name: str, x, diagonal=0, dtype="float32"):
         saveModel(
             name,
             exe,
-            feedkeys=["x"],
+            feed_vars=[node_x],
             fetchlist=[tril_outs],
             inputs=[x],
             outputs=outs,