From 82c1da66f1a8cfabf750e02d92d23fb800c0c561 Mon Sep 17 00:00:00 2001 From: Saaketh Date: Mon, 16 Dec 2024 15:27:43 -0500 Subject: [PATCH] yo --- tests/models/hf/test_hf_base.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/models/hf/test_hf_base.py b/tests/models/hf/test_hf_base.py index bd1ac6f8a8..ec45e31a65 100644 --- 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 \ No newline at end of file + assert n_trainable > 0