Merge remote-tracking branch 'origin/image-generation' into r6763
This commit is contained in:
commit
804589d78b
1 changed files with 6 additions and 1 deletions
|
|
@ -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")
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue