Spaces:

philipp-zettl
/

ChessPT

Sleeping

philipp-zettl commited on Oct 4, 2024

Commit

aa43f32

verified ·

1 Parent(s): f8bdf54

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ model_id = "philipp-zettl/chessPT"
 model_path = hf_hub_download(repo_id=model_id, filename="chessPT.pkl")
 model = DecoderTransformer(vocab_size, n_embed, context_size, n_layer, n_head, dropout)
-model.load_state_dict(torch.load(model_path))
 def greet(prompt):

 model_path = hf_hub_download(repo_id=model_id, filename="chessPT.pkl")
 model = DecoderTransformer(vocab_size, n_embed, context_size, n_layer, n_head, dropout)
+model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
 def greet(prompt):