Updated dev checkpoint w.r.t latest changes in metatrain

README.md

@@ -10,8 +10,8 @@ The model can be downloaded directly from Hugging Face and loaded for use with P
 
 ## List of available models
 
-- PET MAD v1.2.0 (latest)
-- PET MAD v1.1.0
+- PET MAD v1.2.0 (dev)
+- PET MAD v1.1.0 (latest)
 - PET MAD v1.0.1
 - PET MAD v1.0.0
 - PET MAD v0.4.1
@@ -35,7 +35,7 @@ pip install metatrain[pet]
 Then, you can export the model using the following command: 
 
 ```bash
-mtt export https://huggingface.co/lab-cosmo/pet-mad/resolve/main/models/pet-mad-latest.ckpt
+mtt export https://huggingface.co/lab-cosmo/pet-mad/resolve/v1.1.0/models/pet-mad-v1.1.0.ckpt
 ```
 
 ## Usage
@@ -46,7 +46,7 @@ To see the examples of how to use the model, please refer to the PET MAD GitHub
 ## Changelog
 
 - v1.2.0 - Added the uncertainty quantification and ensemble sampling
-- v1.1 - Added the non-conservative forces and stresses
+- v1.1.0 - Added the non-conservative forces and stresses
 - v1.0.1 - Moved to the NativePET backend
 - v1.0 - Initial release of the PET MAD model.
 - v0.4.1 - Technical pre-relase

models/pet-mad-dev.ckpt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:82ceda2d93670a53f6985a5929992bbd54befd4e0670c8f86481378a34838338
-size 19650187
+oid sha256:0bffc062915250742206bd4504492bd0891a3460a4c390a48d6c49e492bbdf00
+size 19649611