Fix model loading (#100)

* match regex to the correct string * update version
evo-design · Dec 17, 2024 · 2670c74 · 2670c74
1 parent 63555e1
commit 2670c74
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/evo/models.py b/evo/models.py
@@ -84,7 +84,7 @@ def load_checkpoint(
     model_config = AutoConfig.from_pretrained(
         hf_model_name,
         trust_remote_code=True,
-        revision='1.1_fix' if re.match(r'evo-1-.*-base', hf_model_name) else 'main',
+        revision='1.1_fix' if re.match(r'evo-1-.*-base', model_name) else 'main',
     )
     model_config.use_cache = True
 
@@ -94,7 +94,7 @@ def load_checkpoint(
         hf_model_name,
         config=model_config,
         trust_remote_code=True,
-        revision='1.1_fix' if re.match(r'evo-1-.*-base', hf_model_name) else 'main',
+        revision='1.1_fix' if re.match(r'evo-1-.*-base', model_name) else 'main',
     )
 
     # Load model state dict & cleanup.

diff --git a/evo/version.py b/evo/version.py
@@ -1 +1 @@
-version = '0.2.1'
+version = '0.3'