AXERA-TECH
/

Speech-Translation.axera

Model card Files Files and versions

qqc1989 commited on Oct 18

Commit

82fcf23

·

verified ·

1 Parent(s): 8ed2455

Update utils/ax_model_bin.py

Files changed (1) hide show

utils/ax_model_bin.py +1 -1

utils/ax_model_bin.py CHANGED Viewed

@@ -76,7 +76,7 @@ class AX_SenseVoiceSmall:
         #     model_file, device_id, intra_op_num_threads=intra_op_num_threads
         # )
         #self.session = axe.InferenceSession(model_file, providers='AxEngineExecutionProvider')
-		self.session = axe.InferenceSession(model_file)
         self.batch_size = batch_size
         self.blank_id = 0
         self.seq_len = seq_len

         #     model_file, device_id, intra_op_num_threads=intra_op_num_threads
         # )
         #self.session = axe.InferenceSession(model_file, providers='AxEngineExecutionProvider')
+        self.session = axe.InferenceSession(model_file)
         self.batch_size = batch_size
         self.blank_id = 0
         self.seq_len = seq_len