minor

convert_sam_hq_to_hf.py

@@ -27,11 +27,7 @@ import re
 import torch
 from safetensors.torch import save_model
 from huggingface_hub import hf_hub_download
-from transformers import (
- SamImageProcessor,
- SamProcessor,
- SamVisionConfig,
-)
+from transformers import SamVisionConfig
 from sam_hq_vit_huge.modeling_sam_hq import SamHQModel
 from sam_hq_vit_huge.configuration_sam_hq import SamHQConfig
 
@@ -132,7 +128,6 @@ def convert_sam_checkpoint(model_name, checkpoint_path, output_dir):
 
  state_dict = torch.load(checkpoint_path, map_location="cpu")
  state_dict = replace_keys(state_dict)
- # print(state_dict.keys())
 
  hf_model = SamHQModel(config)
  hf_model.eval()
@@ -167,6 +162,5 @@ if __name__ == "__main__":
  checkpoint_path = args.checkpoint_path
  else:
  checkpoint_path = hf_hub_download("lkeab/hq-sam", f"{args.model_name}.pth")
- print(checkpoint_path)
 
  convert_sam_checkpoint(args.model_name, checkpoint_path, args.output_dir)