-
Notifications
You must be signed in to change notification settings - Fork 161
/
olive_config.json
127 lines (127 loc) · 7.5 KB
/
olive_config.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
{
"passes": {
"AppendPrePostProcessingOps": {
"module_path": "olive.passes.onnx.append_pre_post_processing_ops.AppendPrePostProcessingOps"
},
"DynamicToFixedShape": { "module_path": "olive.passes.onnx.dynamic_to_fixed_shape.DynamicToFixedShape" },
"ExtractAdapters": { "module_path": "olive.passes.onnx.extract_adapters.ExtractAdapters" },
"ModelBuilder": {
"module_path": "olive.passes.onnx.model_builder.ModelBuilder",
"extra_dependencies": [ "ort-genai" ]
},
"IncDynamicQuantization": {
"module_path": "olive.passes.onnx.inc_quantization.IncDynamicQuantization",
"extra_dependencies": [ "inc" ]
},
"IncQuantization": {
"module_path": "olive.passes.onnx.inc_quantization.IncQuantization",
"extra_dependencies": [ "inc" ]
},
"IncStaticQuantization": {
"module_path": "olive.passes.onnx.inc_quantization.IncStaticQuantization",
"extra_dependencies": [ "inc" ]
},
"InsertBeamSearch": { "module_path": "olive.passes.onnx.insert_beam_search.InsertBeamSearch" },
"MixedPrecisionOverrides": {
"module_path": "olive.passes.onnx.mixed_precision_overrides.MixedPrecisionOverrides"
},
"MoEExpertsDistributor": { "module_path": "olive.passes.onnx.moe_experts_distributor.MoEExpertsDistributor" },
"OnnxBnb4Quantization": { "module_path": "olive.passes.onnx.bnb_quantization.OnnxBnb4Quantization" },
"OnnxConversion": { "module_path": "olive.passes.onnx.conversion.OnnxConversion" },
"OnnxDynamicQuantization": { "module_path": "olive.passes.onnx.quantization.OnnxDynamicQuantization" },
"OnnxFloatToFloat16": { "module_path": "olive.passes.onnx.float16_conversion.OnnxFloatToFloat16" },
"OnnxIOFloat16ToFloat32": { "module_path": "olive.passes.onnx.float32_conversion.OnnxIOFloat16ToFloat32" },
"OnnxMatMul4Quantizer": { "module_path": "olive.passes.onnx.quantization.OnnxMatMul4Quantizer" },
"OnnxModelOptimizer": { "module_path": "olive.passes.onnx.model_optimizer.OnnxModelOptimizer" },
"OnnxOpVersionConversion": { "module_path": "olive.passes.onnx.conversion.OnnxOpVersionConversion" },
"OnnxQuantization": { "module_path": "olive.passes.onnx.quantization.OnnxQuantization" },
"OnnxStaticQuantization": { "module_path": "olive.passes.onnx.quantization.OnnxStaticQuantization" },
"OptimumConversion": {
"module_path": "olive.passes.onnx.optimum_conversion.OptimumConversion",
"extra_dependencies": [ "optimum" ]
},
"OptimumMerging": {
"module_path": "olive.passes.onnx.optimum_merging.OptimumMerging",
"extra_dependencies": [ "optimum" ]
},
"OrtMixedPrecision": { "module_path": "olive.passes.onnx.mixed_precision.OrtMixedPrecision" },
"OrtPerfTuning": {
"module_path": "olive.passes.onnx.perf_tuning.OrtPerfTuning",
"module_dependencies": [ "psutil" ]
},
"OrtTransformersOptimization": {
"module_path": "olive.passes.onnx.transformer_optimization.OrtTransformersOptimization"
},
"QNNPreprocess": { "module_path": "olive.passes.onnx.qnn.qnn_preprocess.QNNPreprocess" },
"VitisAIQuantization": { "module_path": "olive.passes.onnx.vitis_ai_quantization.VitisAIQuantization" },
"VitisQDQQuantizer": { "module_path": "olive.passes.onnx.vitis_ai.quantizer.VitisQDQQuantizer" },
"VitisQOpQuantizer": { "module_path": "olive.passes.onnx.vitis_ai.quantizer.VitisQOpQuantizer" },
"quantize_static": { "module_path": "olive.passes.onnx.vitis_ai.quantize.quantize_static" },
"PowerOfTwoMethod": { "module_path": "olive.passes.onnx.vitis_ai.quant_utils.PowerOfTwoMethod" },
"OpenVINOConversion": {
"module_path": "olive.passes.openvino.conversion.OpenVINOConversion",
"extra_dependencies": [ "openvino" ]
},
"OpenVINOQuantization": {
"module_path": "olive.passes.openvino.quantization.OpenVINOQuantization",
"extra_dependencies": [ "openvino" ]
},
"GptqQuantizer": {
"module_path": "olive.passes.pytorch.gptq.GptqQuantizer",
"module_dependencies": [ "auto-gptq", "optimum" ]
},
"AutoAWQQuantizer": {
"module_path": "olive.passes.pytorch.autoawq.AutoAWQQuantizer",
"module_dependencies": [ "autoawq" ]
},
"MergeAdapterWeights": { "module_path": "olive.passes.pytorch.merge_adapter_weights.MergeAdapterWeights" },
"LoftQ": { "module_path": "olive.passes.pytorch.lora.LoftQ" },
"LoRA": { "module_path": "olive.passes.pytorch.lora.LoRA", "extra_dependencies": [ "lora" ] },
"PyTorchTensorParallel": { "module_path": "olive.passes.pytorch.tensor_parallel.PyTorchTensorParallel" },
"QLoRA": { "module_path": "olive.passes.pytorch.lora.QLoRA", "extra_dependencies": [ "bnb", "lora" ] },
"QuantizationAwareTraining": {
"module_path": "olive.passes.pytorch.quantization_aware_training.QuantizationAwareTraining",
"module_dependencies": [ "pytorch-lightning" ]
},
"SparseGPT": { "module_path": "olive.passes.pytorch.sparsegpt.SparseGPT" },
"SliceGPT": { "module_path": "olive.passes.pytorch.slicegpt.SliceGPT" },
"QuaRot": { "module_path": "olive.passes.pytorch.quarot.QuaRot" },
"TorchTRTConversion": {
"module_path": "olive.passes.pytorch.torch_trt_conversion.TorchTRTConversion",
"extra_dependencies": [ "torch-tensorrt" ]
},
"QNNConversion": { "module_path": "olive.passes.qnn.conversion.QNNConversion" },
"QNNModelLibGenerator": { "module_path": "olive.passes.qnn.model_lib_generator.QNNModelLibGenerator" },
"QNNContextBinaryGenerator": {
"module_path": "olive.passes.qnn.context_binary_generator.QNNContextBinaryGenerator"
},
"SNPEConversion": { "module_path": "olive.passes.snpe.conversion.SNPEConversion" },
"SNPEQuantization": { "module_path": "olive.passes.snpe.quantization.SNPEQuantization" },
"SNPEtoONNXConversion": { "module_path": "olive.passes.snpe.snpe_to_onnx.SNPEtoONNXConversion" },
"NVModelOptQuantization": {
"module_path": "olive.passes.onnx.nvmo_quantization.NVModelOptQuantization",
"extra_dependencies": [ "nvmo" ]
}
},
"extra_dependencies": {
"azureml": [ "azure-ai-ml>=1.11.1", "azure-keyvault-secrets", "azure-identity", "azureml-fsspec" ],
"docker": [ "docker" ],
"cpu": [ "onnxruntime" ],
"gpu": [ "onnxruntime-gpu" ],
"directml": [ "onnxruntime-directml" ],
"openvino": [ "openvino==2023.2.0", "nncf==2.7.0", "numpy<2.0" ],
"tf": [ "tensorflow==1.15.0" ],
"inc": [ "neural-compressor" ],
"optimum": [ "optimum" ],
"torch-tensorrt": [ "torch-tensorrt" ],
"lora": [ "accelerate>=0.30.0", "peft", "scipy" ],
"bnb": [ "bitsandbytes" ],
"ort": [ "onnxruntime", "onnxruntime-directml", "onnxruntime-gpu", "onnxruntime-openvino", "numpy<2.0" ],
"nvmo": [ "nvidia-modelopt~=0.11.0", "onnx-graphsurgeon" ],
"ort-genai": [ "onnxruntime-genai" ],
"capture-onnx-graph": [ "onnxruntime-genai", "optimum" ],
"cloud-cache": [ "azure-storage-blob" ],
"finetune": [ "onnxruntime-genai", "optimum", "accelerate>=0.30.0", "peft", "scipy" ],
"tune-session-params": [ "psutil" ]
}
}