5454 "num_calibration_samples" : 1000 ,
5555
5656 "python_deps" : [
57- [ " ^^" , " python_sync_pip_package" , [[
57+ [ " ^^" , " python_sync_pip_package" , [[
5858 " python_package" ,
59- " package_name=llm-compressor " ,
60- " installable=git+https://github.com/vllm-project/llm-compressor.git@sa/big_model_support " ,
59+ " package_name=compressed-tensors " ,
60+ " package_version=0.5.0 " ,
6161 [" desired_python_version" , [ " ^^" , " get" , " desired_python_version" ] ]
6262 ]] ],
6363 [ " ^^" , " python_sync_pip_package" , [[
6464 " python_package" ,
65- " package_name=compressed-tensors " ,
66- " package_version=0.5.0 " ,
65+ " package_name=llm-compressor " ,
66+ " installable=git+https://github.com/vllm-project/llm-compressor.git@sa/big_model_support " ,
6767 [" desired_python_version" , [ " ^^" , " get" , " desired_python_version" ] ]
6868 ]] ]
6969 ],
8080 "script_extra_params" : [ " ^^" , " substitute" ,
8181 " #{source_model_path}# #{target_model_path}# #{calib_dataset_path}# #{num_gpus}# #{max_sequence_length}# #{num_calibration_samples}#"
8282 ]
83- }
83+ }
0 commit comments