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 8181 - name : Environment
8282 run : |
8383 python utils/print_env.py
84- - name : Slow PyTorch CUDA checkpoint tests on Ubuntu
84+ - name : PyTorch CUDA checkpoint tests on Ubuntu
8585 env :
8686 HF_TOKEN : ${{ secrets.HF_TOKEN }}
8787 # https://pytorch.org/docs/stable/notes/randomness.html#avoiding-nondeterministic-algorithms
@@ -184,7 +184,7 @@ jobs:
184184 run : |
185185 python utils/print_env.py
186186
187- - name : Run slow Flax TPU tests
187+ - name : Run Flax TPU tests
188188 env :
189189 HF_TOKEN : ${{ secrets.HF_TOKEN }}
190190 run : |
@@ -232,7 +232,7 @@ jobs:
232232 run : |
233233 python utils/print_env.py
234234
235- - name : Run slow ONNXRuntime CUDA tests
235+ - name : Run ONNXRuntime CUDA tests
236236 env :
237237 HF_TOKEN : ${{ secrets.HF_TOKEN }}
238238 run : |
Original file line number Diff line number Diff line change 2727from diffusers .utils .testing_utils import (
2828 floats_tensor ,
2929 is_peft_available ,
30+ nightly ,
3031 numpy_cosine_similarity_distance ,
3132 require_peft_backend ,
3233 require_torch_gpu ,
@@ -165,9 +166,10 @@ def test_modify_padding_mode(self):
165166
166167
167168@slow
169+ @nightly
168170@require_torch_gpu
169171@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" )
171173# TODO (DN6, sayakpaul): move these tests to a beefier GPU
172174class FluxLoRAIntegrationTests (unittest .TestCase ):
173175 """internal note: The integration slices were obtained on audace.
Original file line number Diff line number Diff line change 3434from diffusers .utils .import_utils import is_accelerate_available
3535from diffusers .utils .testing_utils import (
3636 load_image ,
37+ nightly ,
3738 numpy_cosine_similarity_distance ,
3839 require_peft_backend ,
3940 require_torch_gpu ,
@@ -207,6 +208,7 @@ def test_integration_move_lora_dora_cpu(self):
207208
208209
209210@slow
211+ @nightly
210212@require_torch_gpu
211213@require_peft_backend
212214class LoraIntegrationTests (unittest .TestCase ):
Original file line number Diff line number Diff line change @@ -113,6 +113,7 @@ def tearDown(self):
113113
114114
115115@slow
116+ @nightly
116117@require_torch_gpu
117118@require_peft_backend
118119class LoraSDXLIntegrationTests (unittest .TestCase ):
You can’t perform that action at this time.
0 commit comments