Merge pull request #55 from evo-design/model_version1.1

Load fixed Evo v1.1 model checkpoint
evo-design · Apr 29, 2024 · 8142b68 · 8142b68
2 parents 9b4fa6d + 1ccba71
commit 8142b68
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/evo/models.py b/evo/models.py
@@ -71,7 +71,11 @@ def load_checkpoint(
 
     # Load model config.
 
-    model_config = AutoConfig.from_pretrained(hf_model_name, trust_remote_code=True)
+    model_config = AutoConfig.from_pretrained(
+        hf_model_name,
+        trust_remote_code=True,
+        revision='1.1_fix',
+    )
     model_config.use_cache = True
 
     # Load model.
@@ -80,6 +84,7 @@ def load_checkpoint(
         hf_model_name,
         config=model_config,
         trust_remote_code=True,
+        revision='1.1_fix',
     )
 
     # Load model state dict & cleanup.

diff --git a/evo/version.py b/evo/version.py
@@ -1 +1 @@
-version = '0.1.1'
+version = '0.1.2'