(reference_root: Path)
| 115 | |
| 116 | |
| 117 | def load_reference_symbols(reference_root: Path) -> tuple[Any, Any, Path]: |
| 118 | use_compatible_transformers() |
| 119 | expected_root = add_reference_root(reference_root) |
| 120 | from vibevoice.modular.modeling_vibevoice_inference import VibeVoiceForConditionalGenerationInference |
| 121 | from vibevoice.processor.vibevoice_processor import VibeVoiceProcessor |
| 122 | import vibevoice |
| 123 | |
| 124 | module_path = Path(vibevoice.__file__).resolve() |
| 125 | try: |
| 126 | module_path.relative_to(expected_root) |
| 127 | except ValueError as exc: |
| 128 | raise RuntimeError(f"VibeVoice imported from {module_path}, expected under {expected_root}") from exc |
| 129 | return VibeVoiceForConditionalGenerationInference, VibeVoiceProcessor, module_path |
| 130 | |
| 131 | |
| 132 | def torch_dtype(name: str) -> torch.dtype: |
no test coverage detected