yasserrmd commited on
Commit
7d3eb57
·
verified ·
1 Parent(s): d6c691f

Update processor/vibevoice_processor.py

Browse files
Files changed (1) hide show
  1. processor/vibevoice_processor.py +3 -1
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(