Spaces:

Tousifahamed
/

smol-lm2-demo

Sleeping

Tousifahamed commited on Jan 22

Commit

7b27885

verified ·

1 Parent(s): adc670e

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ def load_quantized_model(checkpoint_path):
     # Load the checkpoint
     checkpoint = torch.load(checkpoint_path, map_location="cpu")
     # model.load_state_dict(checkpoint["model_state_dict"])
-    model.load_state_dict(checkpoint
     model.eval()
     return model

     # Load the checkpoint
     checkpoint = torch.load(checkpoint_path, map_location="cpu")
     # model.load_state_dict(checkpoint["model_state_dict"])
+    model.load_state_dict(checkpoint)
     model.eval()
     return model