bigger model for now

huggingface · Jul 25, 2024 · 0cb6be7 · 0cb6be7
1 parent 9923084
commit 0cb6be7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/bettertransformer/testing_utils.py b/tests/bettertransformer/testing_utils.py
@@ -27,7 +27,7 @@
 
 MODELS_DICT = {
     "albert": "hf-internal-testing/tiny-random-AlbertModel",
-    "bark": "hf-internal-testing/tiny-random-BarkModel",
+    "bark": "ylacombe/bark-small",
     "bart": "hf-internal-testing/tiny-random-bart",
     "bert": "hf-internal-testing/tiny-random-BertModel",
     "bert-generation": "ybelkada/random-tiny-BertGenerationModel",