rename models save dir

SkyTNT · Oct 6, 2024 · 3f1edd3 · 3f1edd3
1 parent 45ac53e
commit 3f1edd3
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 13 deletions.
diff --git a/.gitignore b/.gitignore
@@ -150,6 +150,7 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 .idea/
-save_models/
+/models/
 output.mid
 /outputs/
+/soundfont.sf2
diff --git a/app_onnx.py b/app_onnx.py
@@ -407,9 +407,9 @@ def get_tokenizer(config_name):
     parser.add_argument("--soundfont-path", type=str, default="soundfont.sf2", help="soundfont")
     parser.add_argument("--model-config", type=str, default="tv2o-medium", help="model config name")
     parser.add_argument("--model-base-path", type=str,
-                        default="save_models/default/model_base.onnx", help="model path")
+                        default="models/default/model_base.onnx", help="model path")
     parser.add_argument("--model-token-path", type=str,
-                        default="save_models/default/model_token.onnx", help="model path")
+                        default="models/default/model_token.onnx", help="model path")
     parser.add_argument("--soundfont-url", type=str,
                         default="https://huggingface.co/skytnt/midi-model/resolve/main/soundfont.sf2",
                         help="download soundfont to soundfont-path if file not exist")
@@ -429,37 +429,37 @@ def get_tokenizer(config_name):
         ],
         "generic pretrain model (tv2o-medium) by skytnt with jpop lora": [
             "tv2o-medium",
-            ["save_models/tv2om_skytnt_jpop_lora/model_base.onnx",
+            ["models/tv2om_skytnt_jpop_lora/model_base.onnx",
              "https://huggingface.co/skytnt/midi-model-tv2om-jpop-lora/resolve/main/onnx/model_base.onnx"],
-            ["save_models/tv2om_skytnt_jpop_lora/model_token.onnx",
+            ["models/tv2om_skytnt_jpop_lora/model_token.onnx",
              "https://huggingface.co/skytnt/midi-model-tv2om-jpop-lora/resolve/main/onnx/model_token.onnx"]
         ],
         "generic pretrain model (tv2o-medium) by skytnt with touhou lora": [
             "tv2o-medium",
-            ["save_models/tv2om_skytnt_touhou_lora/model_base.onnx",
+            ["models/tv2om_skytnt_touhou_lora/model_base.onnx",
              "https://huggingface.co/skytnt/midi-model-tv2om-touhou-lora/resolve/main/onnx/model_base.onnx"],
-            ["save_models/tv2om_skytnt_touhou_lora/model_token.onnx",
+            ["models/tv2om_skytnt_touhou_lora/model_token.onnx",
              "https://huggingface.co/skytnt/midi-model-tv2om-touhou-lora/resolve/main/onnx/model_token.onnx"]
         ],
         "generic pretrain model (tv2o-large) by asigalov61": [
             "tv2o-large",
-            ["save_models/tv2ol_asigalov61/model_base.onnx",
+            ["models/tv2ol_asigalov61/model_base.onnx",
              "https://huggingface.co/asigalov61/Music-Llama/resolve/main/onnx/model_base.onnx"],
-            ["save_models/tv2ol_asigalov61/model_token.onnx",
+            ["models/tv2ol_asigalov61/model_token.onnx",
              "https://huggingface.co/asigalov61/Music-Llama/resolve/main/onnx/model_token.onnx"]
         ],
         "generic pretrain model (tv2o-medium) by asigalov61": [
             "tv2o-medium",
-            ["save_models/tv2om_asigalov61/model_base.onnx",
+            ["models/tv2om_asigalov61/model_base.onnx",
              "https://huggingface.co/asigalov61/Music-Llama-Medium/resolve/main/onnx/model_base.onnx"],
-            ["save_models/tv2om_asigalov61/model_token.onnx",
+            ["models/tv2om_asigalov61/model_token.onnx",
              "https://huggingface.co/asigalov61/Music-Llama-Medium/resolve/main/onnx/model_token.onnx"]
         ],
         "generic pretrain model (tv1-medium) by skytnt": [
             "tv1-medium",
-            ["save_models/tv1m_skytnt/model_base.onnx",
+            ["models/tv1m_skytnt/model_base.onnx",
              "https://huggingface.co/skytnt/midi-model/resolve/main/onnx/model_base.onnx"],
-            ["save_models/tv1m_skytnt/model_token.onnx",
+            ["models/tv1m_skytnt/model_token.onnx",
              "https://huggingface.co/skytnt/midi-model/resolve/main/onnx/model_token.onnx"]
         ]
     }