cocccck / custom_nodes /comfyui-multigpu /web /docs /DownloadAndLoadWav2VecModelMultiGPU.md
bbbboiwow's picture
Upload folder using huggingface_hub
edb09f2 verified
|
Raw
History Blame Contribute Delete
833 Bytes
# DownloadAndLoadWav2VecModelMultiGPU
`DownloadAndLoadWav2VecModelMultiGPU` downloads a preset Wav2Vec2 checkpoint from Hugging Face (if missing) and loads it onto the device you choose, mirroring WanVideo's helper while adding MultiGPU awareness.
## Inputs
### Required
| Parameter | Data Type | Description |
| --- | --- | --- |
| `model` | `STRING` | Preset identifier (`TencentGameMate/chinese-wav2vec2-base` or `facebook/wav2vec2-base-960h`). |
| `base_precision` | `STRING` | Weight precision (`fp32`, `bf16`, `fp16`). |
| `load_device` | `STRING` | Wan loader slot (`main_device` or `offload_device`). |
| `device` | `STRING` | MultiGPU device to run the audio model. |
## Outputs
| Output Name | Data Type | Description |
| --- | --- | --- |
| `wav2vec_model` | `WAV2VECMODEL` | Downloaded and loaded Wav2Vec2 model. |