mirror of
https://github.com/vladmandic/automatic
synced 2026-08-26 06:30:44 +02:00
Vladimir Mandic
7 lines
141 B
Python
7 lines
141 B
Python
import torch
|
|
|
|
|
|
class Generator(torch.Generator):
|
|
def __init__(self, device: torch.device | None = None):
|
|
super().__init__("cpu")
|