Update modeling_sensevoice.py

modeling_sensevoice.py

@@ -1053,8 +1053,19 @@ class SenseVoiceSmall(nn.Module):
 
         return encoder_out, encoder_out_lens
 
+    def export_rebuild_model(model, **kwargs):
+        model.device = kwargs.get("device")
+        model.make_pad_mask = sequence_mask(kwargs["max_seq_len"], flip=False)
+        model.forward = types.MethodType(export_forward, model)
+        model.export_dummy_inputs = types.MethodType(export_dummy_inputs, model)
+        model.export_input_names = types.MethodType(export_input_names, model)
+        model.export_output_names = types.MethodType(export_output_names, model)
+        model.export_dynamic_axes = types.MethodType(export_dynamic_axes, model)
+        model.export_name = types.MethodType(export_name, model)
+        return model
+    
     def export(self, **kwargs):
-        from export_meta import export_rebuild_model
+        # from export_meta import export_rebuild_model
 
         if "max_seq_len" not in kwargs:
             kwargs["max_seq_len"] = 512