@@ -209,6 +209,7 @@ def test_raise_train(self, model_type: str):
209
209
super ()._test_raise_train (model_id )
210
210
211
211
@pytest .mark .gpu_test
212
+ @pytest .mark .accelerate_test
212
213
def test_accelerate_compatibility_cpu_gpu (self ):
213
214
r"""
214
215
Wrapper around the `check_accelerate_compatibility_cpu_gpu` test with `keep_original_model=True`
@@ -217,6 +218,7 @@ def test_accelerate_compatibility_cpu_gpu(self):
217
218
self .check_accelerate_compatibility_cpu_gpu (keep_original_model = True , max_memory = max_memory )
218
219
219
220
@pytest .mark .gpu_test
221
+ @pytest .mark .accelerate_test
220
222
def test_accelerate_compatibility_cpu_gpu_without_keeping (self ):
221
223
r"""
222
224
Wrapper around the `check_accelerate_compatibility_cpu_gpu` test with `keep_original_model=False`
@@ -225,6 +227,7 @@ def test_accelerate_compatibility_cpu_gpu_without_keeping(self):
225
227
self .check_accelerate_compatibility_cpu_gpu (keep_original_model = False , max_memory = max_memory )
226
228
227
229
@pytest .mark .gpu_test
230
+ @pytest .mark .accelerate_test
228
231
def test_accelerate_compatibility_single_gpu (self ):
229
232
r"""
230
233
Wrapper around the `check_accelerate_compatibility_cpu_gpu` test with `keep_original_model=False`
@@ -234,6 +237,7 @@ def test_accelerate_compatibility_single_gpu(self):
234
237
self .check_accelerate_compatibility_cpu_gpu (keep_original_model = True , max_memory = max_memory )
235
238
236
239
@pytest .mark .gpu_test
240
+ @pytest .mark .accelerate_test
237
241
def test_accelerate_compatibility_single_gpu_without_keeping (self ):
238
242
r"""
239
243
Wrapper around the `check_accelerate_compatibility_cpu_gpu` test with `keep_original_model=True`
0 commit comments