From 40d28191c9d9692a441a416c82e7d4389cc8ff33 Mon Sep 17 00:00:00 2001 From: UltralyticsAssistant Date: Fri, 23 Aug 2024 10:41:55 +0000 Subject: [PATCH] Auto-format by https://ultralytics.com/actions --- ultralytics/engine/exporter.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index a412e3e560..5bdd635983 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -1111,7 +1111,7 @@ class Exporter: mct.exporter.pytorch_export_model( model=quant_model_pp, save_model_path=f, repr_dataset=representative_dataset_gen ) - + n_gptq_epochs = 1000 gptq_config = mct.gptq.get_pytorch_gptq_config(n_epochs=n_gptq_epochs, use_hessian_based_weights=False) @@ -1122,9 +1122,10 @@ class Exporter: target_resource_utilization=resource_utilization, gptq_config=gptq_config, core_config=config, - target_platform_capabilities=tpc) + target_platform_capabilities=tpc, + ) - print('Quantized-GPTQ model is ready') + print("Quantized-GPTQ model is ready") return f, None