mirror of
https://github.com/vladmandic/sdnext.git
synced 2026-01-27 15:02:48 +03:00
8 lines
172 B
Python
8 lines
172 B
Python
from typing import Optional
|
|
import torch
|
|
|
|
|
|
class Generator(torch.Generator):
|
|
def __init__(self, device: Optional[torch.device] = None):
|
|
super().__init__("cpu")
|