Spaces:
Running
on
Zero
Running
on
Zero
Update processor/vibevoice_processor.py
Browse files
processor/vibevoice_processor.py
CHANGED
|
@@ -79,7 +79,9 @@ class VibeVoiceProcessor:
|
|
| 79 |
db_normalize = config.get("db_normalize", True)
|
| 80 |
|
| 81 |
# Load tokenizer - try from model path first, then fallback to Qwen
|
| 82 |
-
language_model_pretrained_name = config.get("language_model_pretrained_name", None) or kwargs.pop("language_model_pretrained_name", "Qwen/Qwen2.5-1.5B")
|
|
|
|
|
|
|
| 83 |
logger.info(f"Loading tokenizer from {language_model_pretrained_name}")
|
| 84 |
if 'qwen' in language_model_pretrained_name.lower():
|
| 85 |
tokenizer = VibeVoiceTextTokenizerFast.from_pretrained(
|
|
|
|
| 79 |
db_normalize = config.get("db_normalize", True)
|
| 80 |
|
| 81 |
# Load tokenizer - try from model path first, then fallback to Qwen
|
| 82 |
+
#language_model_pretrained_name = config.get("language_model_pretrained_name", None) or kwargs.pop("language_model_pretrained_name", "Qwen/Qwen2.5-1.5B")
|
| 83 |
+
#Trying Qwen3
|
| 84 |
+
language_model_pretrained_name = kwargs.pop("language_model_pretrained_name", "Qwen/Qwen3-8B")
|
| 85 |
logger.info(f"Loading tokenizer from {language_model_pretrained_name}")
|
| 86 |
if 'qwen' in language_model_pretrained_name.lower():
|
| 87 |
tokenizer = VibeVoiceTextTokenizerFast.from_pretrained(
|