Skip to content

Commit 20553ae

Browse files
committed
comment out gpu testing flag
1 parent 2ef346b commit 20553ae

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

sagemaker-train/tests/integ/train/test_sft_trainer_integration.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ def test_sft_trainer_nova_workflow(sagemaker_session_us_east_1):
137137
assert training_job.output_model_package_arn is not None
138138

139139

140-
@pytest.mark.gpu_intensive
140+
# @pytest.mark.gpu_intensive
141141
def test_sft_trainer_lora_with_sequence_length(sagemaker_session):
142142
"""Test SFT training workflow with LORA and sequence_length specified."""
143143
unique_id = f"{int(time.time())}-{random.randint(1000, 9999)}"

0 commit comments

Comments
 (0)