[diffusion] CI: fix LoRA downloading issues and respect offline flag (#15813)

2025-12-30 11:39:27 +08:00
parent 1e45320198
commit f253f43c9d
4 changed files with 136 additions and 28 deletions
@@ -198,7 +198,12 @@ def run_pytest(files, filter_expr=None):

        is_flaky_ci_assertion = "SafetensorError" in full_output

-        if not (is_perf_assertion or is_flaky_ci_assertion):
+        is_oom_error = (
+            "out of memory" in full_output.lower()
+            or "oom killer" in full_output.lower()
+        )
+
+        if not (is_perf_assertion or is_flaky_ci_assertion or is_oom_error):
            return returncode

    logger.info(f"Max retry exceeded")