Fix utils import issue for nightly tests (#13944)
This commit is contained in:
@@ -14,9 +14,15 @@
|
||||
|
||||
import multiprocessing as mp
|
||||
import os
|
||||
import sys
|
||||
import unittest
|
||||
from pathlib import Path
|
||||
|
||||
from utils import (
|
||||
# Add test directory to path for lora_utils import
|
||||
# TODO: can be removed after migration
|
||||
sys.path.insert(0, str(Path(__file__).resolve().parent.parent.parent))
|
||||
|
||||
from lora_utils import (
|
||||
ALL_OTHER_MULTI_LORA_MODELS,
|
||||
CI_MULTI_LORA_MODELS,
|
||||
run_lora_multiple_batch_on_model_cases,
|
||||
|
||||
Reference in New Issue
Block a user