update

deepjavalibrary · Nov 15, 2024 · a240462 · a240462
1 parent 98af0a3
commit a240462
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/tests/integration/llm/client.py b/tests/integration/llm/client.py
@@ -389,14 +389,14 @@ def get_model_name():
     },
     "gemma-7b-unmerged-lora": {
         "batch_size": [4],
-        "seq_length": [16, 32, 1024],
+        "seq_length": [16, 32],
         "worker": 1,
         "adapters": ["alpaca"],
         "tokenizer": "unsloth/gemma-7b"
     },
     "phi2-unmerged-lora": {
         "batch_size": [4],
-        "seq_length": [16, 32, 1024],
+        "seq_length": [16, 32],
         "worker": 1,
         "adapter_names": ["sql"],
         "tokenizer": "microsoft/phi-2"
@@ -603,7 +603,7 @@ def get_model_name():
     },
     "gemma-7b-unmerged-lora": {
         "batch_size": [4],
-        "seq_length": [16, 32, 1024],
+        "seq_length": [16, 32],
         "worker": 1,
         "adapters": ["alpaca"],
         "tokenizer": "unsloth/gemma-7b"

diff --git a/tests/integration/llm/prepare.py b/tests/integration/llm/prepare.py
@@ -608,11 +608,11 @@
     "phi2-unmerged-lora": {
         "option.model_id": "s3://djl-llm/phi-2/",
         "option.tensor_parallel_degree": "max",
-        "option.adapters": "adapters",
         "option.enable_lora": "true",
         "option.max_loras": 1,
         "option.max_lora_rank": 64,
         "option.long_lora_scaling_factors": "4.0",
+        "option.adapters": "adapters",
         "adapter_ids": [
             "isotr0py/phi-2-test-sql-lora",
         ],
@@ -959,11 +959,11 @@
     "phi2-unmerged-lora": {
         "option.model_id": "s3://djl-llm/phi-2/",
         "option.tensor_parallel_degree": "max",
-        "option.adapters": "adapters",
         "option.enable_lora": "true",
         "option.max_loras": 1,
         "option.max_lora_rank": 64,
         "option.long_lora_scaling_factors": "4.0",
+        "option.adapters": "adapters",
         "adapter_ids": [
             "isotr0py/phi-2-test-sql-lora",
         ],