Procházet zdrojové kódy

Fix "import peft" in tests

Aleksandr Borzunov před 2 roky
rodič
revize
b6e31c6d0f
1 změnil soubory, kde provedl 2 přidání a 1 odebrání
  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: