| | |
|---|
| | | model_dir = tag_name |
|---|
| | | if model_dir.startswith('damo/'): |
|---|
| | | from modelscope.hub.snapshot_download import snapshot_download |
|---|
| | | model_dir = snapshot_download(tag_name, cache_dir=self.cache_dir) |
|---|
| | | model_dir = snapshot_download(model_dir, cache_dir=self.cache_dir) |
|---|
| | | asr_train_config = os.path.join(model_dir, 'config.yaml') |
|---|
| | | asr_model_file = os.path.join(model_dir, 'model.pb') |
|---|
| | | cmvn_file = os.path.join(model_dir, 'am.mvn') |
|---|
| | |
|---|
| | | model, asr_train_args = ASRTask.build_model_from_file( |
|---|
| | | asr_train_config, asr_model_file, cmvn_file, 'cpu' |
|---|
| | | ) |
|---|
| | | self.export(model, tag_name) |
|---|
| | | self._export(model, tag_name) |
|---|
| | | |
|---|
| | | # def export_from_modelscope( |
|---|
| | | # self, |
|---|
