ahatamiz commited on
Commit
13fe540
1 Parent(s): d356f47

Update mamba_vision.py

Browse files
Files changed (1) hide show
  1. mamba_vision.py +0 -17
mamba_vision.py CHANGED
@@ -863,20 +863,3 @@ def mamba_vision_L2(pretrained=False, **kwargs):
863
  torch.hub.download_url_to_file(url=url, dst=model_path)
864
  model._load_state_dict(model_path)
865
  return model
866
-
867
-
868
-
869
- # if __name__ == "__main__":
870
- # import pdb
871
-
872
- # # model = mamba_vision_T()
873
- # # model._load_state_dict('/home/ali/Desktop/Vision_Transformers/hybrid_imagenet/model_weights/mambavision_tiny_1k.pth.tar')
874
- # # model.eval().cuda()
875
- # # model.push_to_hub("nvidia/MambaVision-T-1K")
876
- # model = MambaVision.from_pretrained("nvidia/MambaVision-T-1K")
877
-
878
- # model = AutoModel.from_pretrained("nvidia/MambaVision-T-1K", trust_remote_code=True)
879
-
880
-
881
-
882
- # pdb.set_trace()
 
863
  torch.hub.download_url_to_file(url=url, dst=model_path)
864
  model._load_state_dict(model_path)
865
  return model