Svenni551
/

may-mnist-digits

Model card Files Files and versions Community

Svenni551 commited on Mar 20

Commit

0f4560c

•

1 Parent(s): ebba556

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -62,7 +62,7 @@ class MyNeuralNet(nn.Module):
         x = self.Matrix3(x)
         return x.squeeze()
-model_state_dict = load(hf_hub_download(repo_id="Svenni551/May-stablelm-2-zephyr-1_6b", filename="model.pth"), map_location=torch.device('cpu'))
 model = MyNeuralNet()
 model.load_state_dict(model_state_dict)
 model.eval()

         x = self.Matrix3(x)
         return x.squeeze()
+model_state_dict = load(hf_hub_download(repo_id="Svenni551/may-mnist-digits", filename="model.pth"), map_location=torch.device('cpu'))
 model = MyNeuralNet()
 model.load_state_dict(model_state_dict)
 model.eval()