Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #5078 from Anai-Guo/fix/todo-dead-mps-branch

Browse Source 
fix(todo): drop the unreachable duplicate cuda branch in init_generator
This commit is contained in:
Vladimir Mandic
2026-09-07 05:49:30 +02:00
committed by GitHub
parent ddec927c90 8b738636d0
commit 92e2229cd9
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/todo/todo_merge.py
-2
View File
@@ -29,8 +29,6 @@ def init_generator(device: torch.device, fallback: torch.Generator = None):
return torch.Generator(device="cpu").set_state(torch.get_rng_state())
elif device.type == "cuda":
return torch.Generator(device=device).set_state(torch.cuda.get_rng_state())
elif device.type == "cuda":
return torch.Generator(device=device).set_state(torch.mps.get_rng_state())
else:
if fallback is None:
return init_generator(torch.device("cpu"))