Merge remote-tracking branch 'origin/image-generation' into r6763

This commit is contained in:
Daniel Han 2026-07-27 03:26:28 +00:00
commit 804589d78b

View file

@ -321,7 +321,12 @@ def test_no_light_gpu_init_retry_on_an_accelerator_host(monkeypatch):
attempts = []
class _Blocker:
def find_spec(self, name, path = None, target = None):
def find_spec(
self,
name,
path = None,
target = None,
):
if name == "unsloth_zoo":
attempts.append(os.environ.get("UNSLOTH_ZOO_DISABLE_GPU_INIT"))
raise RuntimeError("CUDA Setup failed despite GPU being available")