File tree Expand file tree Collapse file tree 4 files changed +9
-4
lines changed Expand file tree Collapse file tree 4 files changed +9
-4
lines changed Original file line number Diff line number Diff line change 81
81
- name : Environment
82
82
run : |
83
83
python utils/print_env.py
84
- - name : Slow PyTorch CUDA checkpoint tests on Ubuntu
84
+ - name : PyTorch CUDA checkpoint tests on Ubuntu
85
85
env :
86
86
HF_TOKEN : ${{ secrets.HF_TOKEN }}
87
87
# https://pytorch.org/docs/stable/notes/randomness.html#avoiding-nondeterministic-algorithms
@@ -184,7 +184,7 @@ jobs:
184
184
run : |
185
185
python utils/print_env.py
186
186
187
- - name : Run slow Flax TPU tests
187
+ - name : Run Flax TPU tests
188
188
env :
189
189
HF_TOKEN : ${{ secrets.HF_TOKEN }}
190
190
run : |
@@ -232,7 +232,7 @@ jobs:
232
232
run : |
233
233
python utils/print_env.py
234
234
235
- - name : Run slow ONNXRuntime CUDA tests
235
+ - name : Run ONNXRuntime CUDA tests
236
236
env :
237
237
HF_TOKEN : ${{ secrets.HF_TOKEN }}
238
238
run : |
Original file line number Diff line number Diff line change 27
27
from diffusers .utils .testing_utils import (
28
28
floats_tensor ,
29
29
is_peft_available ,
30
+ nightly ,
30
31
numpy_cosine_similarity_distance ,
31
32
require_peft_backend ,
32
33
require_torch_gpu ,
@@ -165,9 +166,10 @@ def test_modify_padding_mode(self):
165
166
166
167
167
168
@slow
169
+ @nightly
168
170
@require_torch_gpu
169
171
@require_peft_backend
170
- # @unittest.skip("We cannot run inference on this model with the current CI hardware")
172
+ @unittest .skip ("We cannot run inference on this model with the current CI hardware" )
171
173
# TODO (DN6, sayakpaul): move these tests to a beefier GPU
172
174
class FluxLoRAIntegrationTests (unittest .TestCase ):
173
175
"""internal note: The integration slices were obtained on audace.
Original file line number Diff line number Diff line change 34
34
from diffusers .utils .import_utils import is_accelerate_available
35
35
from diffusers .utils .testing_utils import (
36
36
load_image ,
37
+ nightly ,
37
38
numpy_cosine_similarity_distance ,
38
39
require_peft_backend ,
39
40
require_torch_gpu ,
@@ -207,6 +208,7 @@ def test_integration_move_lora_dora_cpu(self):
207
208
208
209
209
210
@slow
211
+ @nightly
210
212
@require_torch_gpu
211
213
@require_peft_backend
212
214
class LoraIntegrationTests (unittest .TestCase ):
Original file line number Diff line number Diff line change @@ -113,6 +113,7 @@ def tearDown(self):
113
113
114
114
115
115
@slow
116
+ @nightly
116
117
@require_torch_gpu
117
118
@require_peft_backend
118
119
class LoraSDXLIntegrationTests (unittest .TestCase ):
You can’t perform that action at this time.
0 commit comments