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

cleanup

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-01-10 08:25:00 -05:00
parent 663b0ed800
commit ac2adaa299
12 changed files with 2 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/todo/todo_merge.py
-1
View File
@@ -25,7 +25,6 @@ def init_generator(device: torch.device, fallback: torch.Generator = None):
"""
Forks the current default random generator given device.
"""
print(f"init_generator device = {device}")
if device.type == "cpu":
return torch.Generator(device="cpu").set_state(torch.get_rng_state())
elif device.type == "cuda":