Skip to content

Commit 7f85be4

Browse files
committed
fix order of decroators.
1 parent 8ab7d17 commit 7f85be4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/models/unets/test_models_unet_2d_condition.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1093,8 +1093,8 @@ def test_wrong_device_map_raises_error(self):
10931093
msg_substring = "You can't pass device_map as a negative int"
10941094
assert msg_substring in str(err_ctx.exception)
10951095

1096-
@require_torch_gpu
10971096
@parameterized.expand([0, "cuda", torch.device("cuda"), torch.device("cuda:0")])
1097+
@require_torch_gpu
10981098
def test_passing_non_dict_device_map_works(self, device_map):
10991099
_, inputs_dict = self.prepare_init_args_and_inputs_for_common()
11001100
loaded_model = self.model_class.from_pretrained(

0 commit comments

Comments
 (0)