Skip to content

Commit 744724b

Browse files
authored
fix: Update Ec2 instance type to g5.4xlarge in test_huggingface_torch_distributed.py (#4181)
This change updates the EC2 instance in the test because g5.8xlarge has lower capacity and frequently throws Capacity errors in us-west-2 region.
1 parent 35c146b commit 744724b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/integ/test_huggingface_torch_distributed.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def test_huggingface_torch_distributed_g5_glue(
3232
transformers_version=huggingface_training_latest_version,
3333
pytorch_version=huggingface_training_pytorch_latest_version,
3434
instance_count=1,
35-
instance_type="ml.g5.8xlarge",
35+
instance_type="ml.g5.4xlarge",
3636
hyperparameters={
3737
"model_name_or_path": "distilbert-base-cased",
3838
"task_name": "wnli",

0 commit comments

Comments
 (0)