Commit ·
700a526
1
Parent(s): 2958ec7
fix: add ubinary to valueerror
Browse files- st_quantize.py +1 -1
st_quantize.py
CHANGED
|
@@ -101,7 +101,7 @@ class FlexibleQuantizer(Module):
|
|
| 101 |
)
|
| 102 |
else:
|
| 103 |
raise ValueError(
|
| 104 |
-
f"Invalid quantization type: {quantization}. Must be 'binary' or 'int8'."
|
| 105 |
)
|
| 106 |
return features
|
| 107 |
|
|
|
|
| 101 |
)
|
| 102 |
else:
|
| 103 |
raise ValueError(
|
| 104 |
+
f"Invalid quantization type: {quantization}. Must be 'binary', 'ubinary', or 'int8'."
|
| 105 |
)
|
| 106 |
return features
|
| 107 |
|