(package: ModelPackage, source: SnapshotSource, models_root: Path, overwrite: bool)
| 1285 | output_weights_file="citrinet_256.safetensors", |
| 1286 | config_kind="citrinet", |
| 1287 | ), |
| 1288 | required_files=( |
| 1289 | "citrinet_256.safetensors", |
| 1290 | "citrinet_256_config.json", |
| 1291 | "citrinet_256_tokenizer.model", |
| 1292 | "citrinet_256_vocab.txt", |
| 1293 | ), |
| 1294 | ), |
| 1295 | ModelPackage( |
| 1296 | id="voxcpm2", |
| 1297 | display_name="VoxCPM2", |
| 1298 | target_directory="VoxCPM2", |
| 1299 | source=SnapshotSource(repo_id="OpenBMB/VoxCPM2"), |
| 1300 | required_files=( |
| 1301 | "config.json", |
| 1302 | "model.safetensors", |
| 1303 | "tokenizer.json", |
| 1304 | "tokenizer_config.json", |
| 1305 | "audiovae.pth", |
| 1306 | "audiovae.safetensors", |
| 1307 | ), |
| 1308 | ), |
| 1309 | ModelPackage( |
| 1310 | id="voxcpm2_audiovae", |
| 1311 | display_name="VoxCPM2 AudioVAE local conversion utility", |
| 1312 | target_directory="VoxCPM2", |
| 1313 | source=ConverterSource( |
| 1314 | kind="pytorch_to_safetensors", |
| 1315 | description="Convert a local PyTorch checkpoint into safetensors.", |
| 1316 | ), |
no test coverage detected