fix: allow config.json and tokenizer files for DiffRhythm models

This commit is contained in:
2025-11-24 12:26:33 +01:00
parent bf5c150d23
commit c31b1e87bd

View File

@@ -394,8 +394,12 @@ for file_path in latest_snapshot.rglob('*'):
# If filename filter is specified, only match those files
if filename_filter and filename_filter not in file_path.name:
continue
# Skip metadata files
if file_path.name.endswith(('.json', '.txt', '.md', '.gitattributes')):
# Skip metadata files except important config files
# Allow: config.json, tokenizer.json, tokenizer_config.json, sentencepiece models
important_files = ('config.json', 'tokenizer.json', 'tokenizer_config.json', '.model')
if file_path.name.endswith(('.txt', '.md', '.gitattributes')):
continue
if file_path.name.endswith('.json') and not file_path.name in important_files:
continue
print(str(file_path))
EOPYFINDFIND