😝 Add test config save_pretrained.

TensorSpeech · May 15, 2021 · 77f6ac5 · 77f6ac5
1 parent 82acd37
commit 77f6ac5
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 55 deletions.
diff --git a/tacotron2/config.yml b/tacotron2/config.yml
diff --git a/tacotron2/processor.json b/tacotron2/processor.json
diff --git a/test/test_auto.py b/test/test_auto.py
@@ -66,3 +66,6 @@ def test_auto_processor(mapper_path):
 def test_auto_model(config_path):
     config = AutoConfig.from_pretrained(pretrained_path=config_path)
     model = TFAutoModel.from_pretrained(config=config, pretrained_path=None)
+
+    # test save_pretrained
+    config.save_pretrained("./")