File tree 2 files changed +12
-12
lines changed
2 files changed +12
-12
lines changed Original file line number Diff line number Diff line change @@ -366,13 +366,13 @@ def run(self):
366
366
q_config = prepare_q_config (self .args )
367
367
q_config ["activation_format" ] = "f8e4m3"
368
368
369
- quantization_config = dict (
370
- weight_quantization_config = wc_config ,
371
- activation_quantization_config = q_config ,
372
- num_samples = self .args .num_samples ,
373
- dataset = self .args .dataset ,
374
- trust_remote_code = self .args .trust_remote_code ,
375
- )
369
+ quantization_config = {
370
+ " weight_quantization_config" : wc_config ,
371
+ " activation_quantization_config" : q_config ,
372
+ " num_samples" : self .args .num_samples ,
373
+ " dataset" : self .args .dataset ,
374
+ " trust_remote_code" : self .args .trust_remote_code ,
375
+ }
376
376
else :
377
377
quantization_config = prepare_q_config (self .args )
378
378
ov_config = OVConfig (quantization_config = quantization_config )
Original file line number Diff line number Diff line change @@ -890,11 +890,11 @@ def __init__(
890
890
# Pull dataset-related parameters from child configs. This is not the intended use case, but we process it just
891
891
# in case user sets those parameters inside child configs only.
892
892
wqc , aqc = self .weight_quantization_config , self .activation_quantization_config
893
- num_samples = num_samples or self . wqc .num_samples or self . aqc .num_samples
894
- dataset = dataset or self . wqc .dataset or self . aqc .dataset
895
- tokenizer = tokenizer or self . wqc .tokenizer or self . aqc .tokenizer
896
- processor = processor or self . wqc .processor or self . aqc .processor
897
- trust_remote_code = trust_remote_code or self . wqc .trust_remote_code or self . aqc .trust_remote_code
893
+ num_samples = num_samples or wqc .num_samples or aqc .num_samples
894
+ dataset = dataset or wqc .dataset or aqc .dataset
895
+ tokenizer = tokenizer or wqc .tokenizer or aqc .tokenizer
896
+ processor = processor or wqc .processor or aqc .processor
897
+ trust_remote_code = trust_remote_code or wqc .trust_remote_code or aqc .trust_remote_code
898
898
super ().__init__ (
899
899
num_samples = num_samples ,
900
900
dataset = dataset ,
You can’t perform that action at this time.
0 commit comments