nvidia
/

MambaVision-T-1K

Image Feature Extraction

image-classification

Model card Files Files and versions Community

ahatamiz commited on about 1 month ago

Commit

4f1cf78

•

1 Parent(s): eb35522

Update hf_model.py

Files changed (1) hide show

hf_model.py +0 -1

hf_model.py CHANGED Viewed

@@ -47,7 +47,6 @@ class MambaVisionModel(PreTrainedModel):
         args = MambaVisionArgs(**config.args)
         self.config = config
         self.model = create_model(args.model)
-        load_checkpoint(self.model, args.checkpoint)
         self.model.load_state_dict(torch.load("pytorch_model.bin"))
     def forward(self, x: torch.Tensor):

         args = MambaVisionArgs(**config.args)
         self.config = config
         self.model = create_model(args.model)
         self.model.load_state_dict(torch.load("pytorch_model.bin"))
     def forward(self, x: torch.Tensor):