Fix "import peft" in tests

9 months ago · b6e31c6d0f
parent d8298faa00
commit b6e31c6d0f
1 changed files with 2 additions and 1 deletions
--- a/tests/test_full_model.py
+++ b/tests/test_full_model.py
@ -1,4 +1,3 @@
-import peft
 import pytest
 import torch
 import transformers
@ -67,6 +66,8 @@ def test_full_model_exact_match(use_peft: bool, pass_empty_tensors: bool, atol_f
                REF_NAME, low_cpu_mem_usage=True, torch_dtype=torch.float32
            )
            if use_peft:
+                import peft
+
                ref_model = peft.PeftModel.from_pretrained(ref_model, ADAPTER_NAME)
                ref_model.train(False)
            if config.vocab_size < ref_model.config.vocab_size: