mkrimmel-pplx commited on
Commit
700a526
·
1 Parent(s): 2958ec7

fix: add ubinary to valueerror

Browse files
Files changed (1) hide show
  1. 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