devignite commited on
Commit
a71596d
·
verified ·
1 Parent(s): 64b5604

Enhance quantization check for model name to include variations

Browse files
Files changed (1) hide show
  1. nodes/base_vibevoice.py +1 -1
nodes/base_vibevoice.py CHANGED
@@ -284,7 +284,7 @@ class BaseVibeVoiceNode:
284
 
285
  # Check if this is a quantized model based on the model name
286
  is_quantized_4bit = "Quant-4Bit" in model_name
287
- is_quantized_8bit = "Quant-8Bit" in model_name # Future support
288
 
289
  # Prepare attention implementation kwargs
290
  model_kwargs = {
 
284
 
285
  # Check if this is a quantized model based on the model name
286
  is_quantized_4bit = "Quant-4Bit" in model_name
287
+ is_quantized_8bit = "Quant-8Bit" in model_name or "Q8" in model_name or "8bit" in model_name.lower()
288
 
289
  # Prepare attention implementation kwargs
290
  model_kwargs = {