Compare commits

...

1 Commits

Author SHA1 Message Date
sayakpaul
ac1a6eca5c start fixing nan 2024-10-18 14:39:08 +05:30

View File

@@ -167,7 +167,7 @@ class FluxLoRATests(unittest.TestCase, PeftLoraLoaderMixinTests):
@slow
@require_torch_gpu
@require_peft_backend
# @unittest.skip("We cannot run inference on this model with the current CI hardware")
@unittest.skip("We cannot run inference on this model with the current CI hardware")
# TODO (DN6, sayakpaul): move these tests to a beefier GPU
class FluxLoRAIntegrationTests(unittest.TestCase):
"""internal note: The integration slices were obtained on audace.