Spaces:
Running
on
Zero
Running
on
Zero
Enhance quantization check for model name to include variations
Browse files- 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
|
| 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 = {
|