From f0fd2633347fdfc829bb559b77b08a17c1febfb8 Mon Sep 17 00:00:00 2001 From: vasiliy Date: Mon, 6 Oct 2025 04:33:10 -0700 Subject: [PATCH] small cleanup Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags: --- hf_torchao_vllm/inspect_llm_compressor_output.py | 9 --------- 1 file changed, 9 deletions(-) diff --git a/hf_torchao_vllm/inspect_llm_compressor_output.py b/hf_torchao_vllm/inspect_llm_compressor_output.py index 9416593..75cb693 100644 --- a/hf_torchao_vllm/inspect_llm_compressor_output.py +++ b/hf_torchao_vllm/inspect_llm_compressor_output.py @@ -19,14 +19,5 @@ def run( model_name, model_extension = 'model', 'safetensors' inspect_model_state_dict(dir_name, model_name, model_extension) - if False: - # inpect the model, saved in safetensors format - model_name = f'{dir_name}/model.safetensors' - with safetensors.safe_open(model_name, framework='pt', device='cpu') as f: - print(f.metadata()) - for k in f.keys(): - t = f.get_tensor(k) - print(k, t.shape, t.dtype) - if __name__ == '__main__': fire.Fire(run)