[Minor] Change lora_target_modules to "all" in CI tests (#17386)
This commit is contained in:
@@ -29,15 +29,7 @@ class TestLoRALoadFromTensor(CustomTestCase):
|
||||
model_path=MODEL_PATH,
|
||||
enable_lora=True,
|
||||
max_lora_rank=64,
|
||||
lora_target_modules=[
|
||||
"q_proj",
|
||||
"k_proj",
|
||||
"v_proj",
|
||||
"o_proj",
|
||||
"gate_proj",
|
||||
"up_proj",
|
||||
"down_proj",
|
||||
],
|
||||
lora_target_modules=["all"],
|
||||
mem_fraction_static=0.6,
|
||||
log_level="error",
|
||||
)
|
||||
@@ -61,15 +53,7 @@ class TestLoRALoadFromTensor(CustomTestCase):
|
||||
model_path=MODEL_PATH,
|
||||
enable_lora=True,
|
||||
max_lora_rank=64,
|
||||
lora_target_modules=[
|
||||
"q_proj",
|
||||
"k_proj",
|
||||
"v_proj",
|
||||
"o_proj",
|
||||
"gate_proj",
|
||||
"up_proj",
|
||||
"down_proj",
|
||||
],
|
||||
lora_target_modules=["all"],
|
||||
mem_fraction_static=0.6,
|
||||
log_level="error",
|
||||
max_loaded_loras=MAX_LOADED_LORAS,
|
||||
|
||||
Reference in New Issue
Block a user