mcrimi
/

snakeformer

game-simulation

Model card Files Files and versions

mcrimi commited on Jan 31

Commit

4b7659f

·

verified ·

1 Parent(s): 99e749a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -155,7 +155,7 @@ config = GPTConfig(
     n_layer=meta.get("n_layer", 4),
 )
 model = GPT(config)
-model.load_state_dict(torch.load(model_path, map_location="cpu"))
 model.eval()
 ```

     n_layer=meta.get("n_layer", 4),
 )
 model = GPT(config)
+model.load_state_dict(torch.load(model_path))
 model.eval()
 ```