Add amitaro live model

2024-06-01 23:54:58 +09:00
parent d31cd0da38
commit bbd89794c6
2 changed files with 7 additions and 2 deletions
--- a/initialize.py
+++ b/initialize.py
@@ -78,7 +78,12 @@ def download_default_models():
            "koharune-ami/config.json",
            "koharune-ami/style_vectors.npy",
            "koharune-ami/koharune-ami.safetensors",
-        ]
+        ],
+        "litagin/sbv2_amitaro": [
+            "amitaro/config.json",
+            "amitaro/style_vectors.npy",
+            "amitaro/amitaro.safetensors",
+        ],
    }
    for repo_id, files in additional_files.items():
        for file in files: