| 948 | required_files=( |
| 949 | "flow/config.yaml", |
| 950 | "flow/flow.pt", |
| 951 | "hift/hift.pt", |
| 952 | "llm/config.json", |
| 953 | "llm/generation_config.json", |
| 954 | "llm/model-00001-of-00002.safetensors", |
| 955 | "llm/model-00002-of-00002.safetensors", |
| 956 | "llm/model.safetensors.index.json", |
| 957 | "speech_tokenizer/config.json", |
| 958 | "speech_tokenizer/model.safetensors", |
| 959 | "speech_tokenizer/preprocessor_config.json", |
| 960 | "vq32k-phoneme-tokenizer/tokenizer.model", |
| 961 | "vq32k-phoneme-tokenizer/tokenizer_config.json", |
| 962 | ), |
| 963 | ), |
| 964 | SnapshotPlacement( |
| 965 | source=SnapshotSource( |
| 966 | repo_id="mlx-community/index-tts2-mlx", |
| 967 | include_prefixes=("campplus.safetensors",), |
| 968 | ), |
| 969 | target_subdir="frontend", |
| 970 | required_files=("campplus.safetensors",), |
| 971 | ), |
| 972 | ), |
| 973 | ), |
| 974 | required_files=( |
| 975 | "audio_cpp_config.json", |