scheduler_config

huggingface · Dec 12, 2024 · c168e23 · c168e23
1 parent 360ddd1
commit c168e23
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/huggingface_hub/serialization/_dduf.py b/src/huggingface_hub/serialization/_dduf.py
@@ -326,7 +326,7 @@ def _validate_dduf_structure(index: Any, entry_names: Iterable[str]) -> None:
     Rules:
     - The 'model_index.json' entry is required and must contain a dictionary.
     - Each folder name must correspond to an entry in 'model_index.json'.
-    - Each folder must contain at least a config file ('config.json', 'tokenizer_config.json', 'image_processor.json').
+    - Each folder must contain at least a config file ('config.json', 'tokenizer_config.json', 'image_processor.json', 'scheduler_config.json').
 
     Args:
         index (Any):

diff --git a/tests/test_dduf.py b/tests/test_dduf.py
@@ -103,7 +103,7 @@ def test_validate_dduf_structure_missing_config_file(self):
             _validate_dduf_structure(
                 {"encoder": {}},
                 {
-                    "encoder/not_a_config.json",  # expecting a config.json / tokenizer_config.json / image_processor.json
+                    "encoder/not_a_config.json",  # expecting a config.json / tokenizer_config.json / image_processor.json / scheduler_config.json
                     "encoder/model.safetensors",
                 },
             )