Skip to content

Commit ca98462

Browse files
authored
Merge pull request #70 from vkuzo/20251006_cleanup
small cleanup
2 parents 8cfca15 + f0fd263 commit ca98462

File tree

1 file changed

+0
-9
lines changed

1 file changed

+0
-9
lines changed

hf_torchao_vllm/inspect_llm_compressor_output.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -19,14 +19,5 @@ def run(
1919
model_name, model_extension = 'model', 'safetensors'
2020
inspect_model_state_dict(dir_name, model_name, model_extension)
2121

22-
if False:
23-
# inpect the model, saved in safetensors format
24-
model_name = f'{dir_name}/model.safetensors'
25-
with safetensors.safe_open(model_name, framework='pt', device='cpu') as f:
26-
print(f.metadata())
27-
for k in f.keys():
28-
t = f.get_tensor(k)
29-
print(k, t.shape, t.dtype)
30-
3122
if __name__ == '__main__':
3223
fire.Fire(run)

0 commit comments

Comments
 (0)