Make bark could have tiny model (#25290)

* temp * update * update * update * small dim * small dim * small dim * fix * update * fix * fix * fix * fix * fix * fix * fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2023-08-04 15:13:14 +02:00
parent f0fd73a2de
commit ce6d153a53
3 changed files with 95 additions and 26 deletions
--- a/utils/create_dummy_models.py
+++ b/utils/create_dummy_models.py
@@ -974,6 +974,10 @@ def get_token_id_from_tokenizer(token_id_name, tokenizer, original_token_id):


 def get_config_overrides(config_class, processors):
+    # `Bark` configuration is too special. Let's just not handle this for now.
+    if config_class.__name__ == "BarkConfig":
+        return {}
+
    config_overrides = {}

    # Check if there is any tokenizer (prefer fast version if any)