yo

mosaicml · Dec 16, 2024 · 82c1da6 · 82c1da6
1 parent 7b12371
commit 82c1da6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/models/hf/test_hf_base.py b/tests/models/hf/test_hf_base.py
@@ -2,9 +2,10 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import pytest
-from llmfoundry.models.hf.hf_base import BaseHuggingFaceModel
 from peft import PeftModel
 
+from llmfoundry.models.hf.hf_base import BaseHuggingFaceModel
+
 
 def test_build_inner_model_fsdp():
     model = BaseHuggingFaceModel.build_inner_model(
@@ -45,4 +46,4 @@ def test_pretrained_peft_trainable():
 
     n_trainable, n_all = model.get_nb_trainable_parameters()
     assert n_all > 0
-    assert n_trainable > 0
+    assert n_trainable > 0