Skip to content

Commit

Permalink
[ci] remove partition steps for deepspeed/hf based models (#1524)
Browse files Browse the repository at this point in the history
  • Loading branch information
siddvenk authored Jan 30, 2024
1 parent f3f4cfd commit 32d4b8d
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions tests/integration/llm/sagemaker-endpoint-tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,6 @@ def boolean_arg(value):
"dtype": "fp32",
"number_of_partitions": 1,
},
"partition": True,
"cls_to_use": HuggingFaceAccelerateModel,
},
"gpt-j-6b": {
Expand All @@ -76,7 +75,6 @@ def boolean_arg(value):
"tensor_parallel_degree": 2,
"parallel_loading": True,
},
"partition": True,
"cls_to_use": DeepSpeedModel,
},
"pythia-12b": {
Expand All @@ -86,7 +84,6 @@ def boolean_arg(value):
"tensor_parallel_degree": 4,
"parallel_loading": True,
},
"partition": True,
"partition_s3_uri": "s3://djl-llm-sm-endpoint-tests/pythia-12b-4p/",
"cls_to_use": DeepSpeedModel,
}
Expand Down

0 comments on commit 32d4b8d

Please sign in to comment.