minor

plutonium-239 · Apr 1, 2024 · 1911520 · 1911520
1 parent ce73898
commit 1911520
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/makefile b/makefile
@@ -31,7 +31,7 @@ ruff:
 .PHONY: test test-full
 
 test:
-	@pytest -vx --cov=memsave_torch test -m quick
+	@pytest -vx --cov=memsave_torch/nn test -m quick
 
 test-full:
 	@pytest -vx --cov=memsave_torch test
diff --git a/memsave_torch/nn/LayerNorm.py b/memsave_torch/nn/LayerNorm.py
@@ -83,6 +83,8 @@ def from_nn_LayerNorm(cls, ln: nn.LayerNorm):
                 f"Trying to load a model saved in torch>=2.1, but system version is {torch_version}. \n"
                 + "This is problematic because torch 2.1 changed how LayerNorm bias works."
             )
+        else:
+            obj.bias = ln.bias
         return obj