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)