From b6e31c6d0f828f100fdf818fae85d4fe98831bfb Mon Sep 17 00:00:00 2001
From: Aleksandr Borzunov <hxrussia@gmail.com>
Date: Sat, 5 Aug 2023 22:23:58 +0000
Subject: [PATCH] Fix "import peft" in tests

---
 tests/test_full_model.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/test_full_model.py b/tests/test_full_model.py
index 511604b..903f685 100644
--- 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: