File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
examples/models/llama/config Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -162,11 +162,11 @@ class QuantizationConfig:
162162 pt2e_quantize : Optional [Pt2eQuantize ] = None
163163 group_size : Optional [int ] = None
164164 use_spin_quant : Optional [SpinQuant ] = None
165- use_qat : Optional [ bool ] = None
165+ use_qat : bool = False
166166 calibration_tasks : Optional [List [str ]] = None
167167 calibration_limit : Optional [int ] = None
168168 calibration_seq_length : Optional [int ] = None
169- calibration_data : Optional [ str ] = None
169+ calibration_data : str = "Once upon a time"
170170
171171 def __post_init__ (self ):
172172 if self .qmode :
@@ -243,7 +243,7 @@ class QNNConfig:
243243
244244@dataclass
245245class MPSConfig :
246- enabled : Optional [ bool ] = False
246+ enabled : bool = False
247247
248248
249249@dataclass
You can’t perform that action at this time.
0 commit comments