We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2ef346b commit 20553aeCopy full SHA for 20553ae
1 file changed
sagemaker-train/tests/integ/train/test_sft_trainer_integration.py
@@ -137,7 +137,7 @@ def test_sft_trainer_nova_workflow(sagemaker_session_us_east_1):
137
assert training_job.output_model_package_arn is not None
138
139
140
-@pytest.mark.gpu_intensive
+# @pytest.mark.gpu_intensive
141
def test_sft_trainer_lora_with_sequence_length(sagemaker_session):
142
"""Test SFT training workflow with LORA and sequence_length specified."""
143
unique_id = f"{int(time.time())}-{random.randint(1000, 9999)}"
0 commit comments