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:
162
162
pt2e_quantize : Optional [Pt2eQuantize ] = None
163
163
group_size : Optional [int ] = None
164
164
use_spin_quant : Optional [SpinQuant ] = None
165
- use_qat : Optional [ bool ] = None
165
+ use_qat : bool = False
166
166
calibration_tasks : Optional [List [str ]] = None
167
167
calibration_limit : Optional [int ] = None
168
168
calibration_seq_length : Optional [int ] = None
169
- calibration_data : Optional [ str ] = None
169
+ calibration_data : str = "Once upon a time"
170
170
171
171
def __post_init__ (self ):
172
172
if self .qmode :
@@ -243,7 +243,7 @@ class QNNConfig:
243
243
244
244
@dataclass
245
245
class MPSConfig :
246
- enabled : Optional [ bool ] = False
246
+ enabled : bool = False
247
247
248
248
249
249
@dataclass
You can’t perform that action at this time.
0 commit comments