Update hf_model.py
Browse files- hf_model.py +0 -1
hf_model.py
CHANGED
|
@@ -47,7 +47,6 @@ class MambaVisionModel(PreTrainedModel):
|
|
| 47 |
args = MambaVisionArgs(**config.args)
|
| 48 |
self.config = config
|
| 49 |
self.model = create_model(args.model)
|
| 50 |
-
self.model.load_state_dict(torch.load("pytorch_model.bin"))
|
| 51 |
|
| 52 |
def forward(self, x: torch.Tensor):
|
| 53 |
return self.model.forward(x)
|
|
|
|
| 47 |
args = MambaVisionArgs(**config.args)
|
| 48 |
self.config = config
|
| 49 |
self.model = create_model(args.model)
|
|
|
|
| 50 |
|
| 51 |
def forward(self, x: torch.Tensor):
|
| 52 |
return self.model.forward(x)
|