nvidia
/

MambaVision-T-1K

Image Feature Extraction

image-classification

Model card Files Files and versions Community

ahatamiz commited on about 1 month ago

Commit

43434c2

•

1 Parent(s): 4f1cf78

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)
-        self.model.load_state_dict(torch.load("pytorch_model.bin"))
     def forward(self, x: torch.Tensor):
         return self.model.forward(x)

         args = MambaVisionArgs(**config.args)
         self.config = config
         self.model = create_model(args.model)
     def forward(self, x: torch.Tensor):
         return self.model.forward(x)