Update mamba_vision.py
Browse files- 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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|