소스 검색

Fix "import peft" in tests

Aleksandr Borzunov 2 년 전
부모
커밋
b6e31c6d0f
1개의 변경된 파일2개의 추가작업 그리고 1개의 파일을 삭제
  1. 2 1
      tests/test_full_model.py

+ 2 - 1
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: