From dbb60fda41f74ea2e57da33aee06869da29dc876 Mon Sep 17 00:00:00 2001
From: anniewesterlund <34099005+anniewesterlund@users.noreply.github.com>
Date: Thu, 15 Feb 2024 09:43:56 +0100
Subject: [PATCH] Update README.md

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 9d504d7..4ebcb5a 100644
--- a/README.md
+++ b/README.md
@@ -7,7 +7,7 @@ The pre-trained model was tested on downstream tasks such as reaction prediction
 The public models and datasets available [here](https://az.box.com/s/7eci3nd9vy0xplqniitpk02rbg9q2zcq).  To run these models with the new version, you first need to update the checkpoint, e.g.:
 ```
 model = torch.load("model.ckpt")
-model["hyper_parameters"]["vocabulary_size"] = model.pop("vocab_size")
+model["hyper_parameters"]["vocabulary_size"] = model["hyper_parameters"].pop("vocab_size")
 torch.save(model, "model_v2.ckpt")
 ```