Skip to content

Commit b5eaedb

Browse files
authored
Merge pull request #78 from foundation-model-stack/encoder_fix
Update backend to `sendnn`
2 parents 596baca + ab870c0 commit b5eaedb

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

scripts/run_encoders.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,10 @@
9090
if args.compile:
9191
dprint("Compiling model...")
9292
if args.is_aiu_backend:
93-
model.compile(backend="sendnn_decoder")
93+
model.compile(
94+
backend="sendnn",
95+
options={'sendnn.dynamic': args.compile_dynamic_sendnn},
96+
)
9497
else:
9598
# compiling can make first inference pass slow
9699
model.compile(mode=args.compile_mode, backend=args.compile_backend)

0 commit comments

Comments
 (0)