Skip to content

Commit

Permalink
fix awq exporting (#358)
Browse files Browse the repository at this point in the history
  • Loading branch information
wenhuach21 authored Dec 2, 2024
1 parent 174b7cc commit ca9e655
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion auto_round/export/export_to_awq/export.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,7 +123,7 @@ def save_quantized_as_autoawq(output_dir, inplace=True, **kwargs):
with ThreadPoolExecutor(max_workers=2) as executor:
with tqdm(total=len(names), leave=True) as pbar:
def wrapper(name):
pack_layer(name, model, layer_config, backend, pbar)
pack_layer(name, compressed_model, layer_config, backend, pbar)

for _ in executor.map(wrapper, names):
pass
Expand Down

0 comments on commit ca9e655

Please sign in to comment.