compatibility with older transformers version yl4579#36

zolero · Nov 19, 2023 · 5182a4a · 5182a4a
1 parent dbbc7e5
commit 5182a4a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models.py b/models.py
@@ -699,7 +699,7 @@ def load_checkpoint(model, optimizer, path, load_only_params=True, ignore_module
     for key in model:
         if key in params and key not in ignore_modules:
             print('%s loaded' % key)
-            model[key].load_state_dict(params[key])
+            model[key].load_state_dict(params[key], strict=False)
     _ = [model[key].eval() for key in model]
 
     if not load_only_params: