Views
No views yet
1from vibevoice.modular.modeling_vibevoice_inference import VibeVoiceForConditionalGenerationInference
2from vibevoice.processor.vibevoice_processor import VibeVoiceProcessor
3
4# For 4-bit model
5model = VibeVoiceForConditionalGenerationInference.from_pretrained(
6 "Dannidee/VibeVoice7b-low-vram/4bit",
7 device_map='cuda',
8 torch_dtype=torch.bfloat16,
9)
10processor = VibeVoiceProcessor.from_pretrained("Dannidee/VibeVoice7b-low-vram/4bit")