From af14f54bc6d9484a6f72e9e256cbd9a26c86e495 Mon Sep 17 00:00:00 2001 From: AllentDan Date: Tue, 14 Jan 2025 16:23:40 +0800 Subject: [PATCH] remove duplicate print --- lmdeploy/lite/quantization/awq.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/lmdeploy/lite/quantization/awq.py b/lmdeploy/lite/quantization/awq.py index 59a7f8ec6..7fd4de431 100644 --- a/lmdeploy/lite/quantization/awq.py +++ b/lmdeploy/lite/quantization/awq.py @@ -359,7 +359,6 @@ def smooth_layers(layers, max_memory = torch.cuda.max_memory_allocated() / 1024 / 1024 / 1024 print(f'{l_name} smooth weight done.' f' max gpu memory: {max_memory:.2f} GB') - print(f'{l_name} smooth weight done.') def pseudo_quantize_tensor(w, @@ -431,4 +430,3 @@ def awq_layers(layers, max_memory = torch.cuda.max_memory_allocated() / 1024 / 1024 / 1024 print(f'{l_name} smooth weight done.' f' max gpu memory: {max_memory:.2f} GB') - print(f'{l_name} smooth weight done.')