File size: 691 Bytes
7d52396 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
import sys
from pathlib import Path
import pytest
wd = Path(__file__).parent.parent.absolute()
@pytest.fixture()
def orig_llama():
sys.path.append(str(wd))
from scripts.download import download_original
download_original(wd)
import original_model
return original_model
@pytest.fixture()
def orig_llama_adapter():
sys.path.append(str(wd))
from scripts.download import download_original
download_original(wd)
import original_adapter
return original_adapter
@pytest.fixture()
def lit_llama():
# this adds support for running tests without the package installed
sys.path.append(str(wd))
import lit_llama
return lit_llama
|