Views
No views yet
1from vibevoice.modular.modeling_vibevoice_inference import VibeVoiceForConditionalGenerationInference
2from vibevoice.processor.vibevoice_processor import VibeVoiceProcessor
3
4# Load with f16 precision
5model = VibeVoiceForConditionalGenerationInference.from_pretrained(
6 "./VibeVoice-Large-f16",
7 torch_dtype=torch.float16,
8 device_map="cpu"
9)
10
11processor = VibeVoiceProcessor.from_pretrained("./VibeVoice-Large-f16")
12
13# Use --use_f16 flag with demo scripts
14python demo/inference_from_file.py --model_path ./VibeVoice-Large-f16 --use_f16 --device cpu