File tree 1 file changed +6
-0
lines changed 1 file changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -690,6 +690,9 @@ def recurse_elems(elem: Any):
690
690
"quant_method" : "fp8" ,
691
691
"activation_scheme" : "static"
692
692
}
693
+ elif quantization .get ("quant_method" ) == "compressed-tensors" :
694
+ # Pass through the quantization config to compressed-tensors
695
+ quantization_config = quantization
693
696
else :
694
697
raise ValueError (
695
698
f"Found unknown quantization='{ quantization } ' in config" )
@@ -707,13 +710,16 @@ def recurse_elems(elem: Any):
707
710
708
711
if config_type == "multimodal" :
709
712
multimodal_config = config_dict .pop ("vision_encoder" )
713
+ quantization_config = config_dict .get ("quantization_config" , {})
710
714
711
715
config_dict = {
712
716
"text_config" : config_dict ,
713
717
"vision_config" : multimodal_config
714
718
}
715
719
config_dict ["architectures" ] = ["PixtralForConditionalGeneration" ]
716
720
config_dict ["model_type" ] = "pixtral"
721
+ if quantization_config :
722
+ config_dict ["quantization_config" ] = quantization_config
717
723
718
724
config_dict .update (kwargs )
719
725
You can’t perform that action at this time.
0 commit comments