mirror of
https://github.com/vladmandic/automatic
synced 2026-08-26 15:16:01 +02:00
Seunghoon Lee
7 lines
171 B
Python
7 lines
171 B
Python
import torch
|
|
from typing import Optional
|
|
|
|
class Generator(torch.Generator):
|
|
def __init__(self, device: Optional[torch.device] = None):
|
|
super().__init__("cpu")
|