c19ca42
1
2
3
4
5
6
7
8
from typing import Optional import torch class Generator(torch.Generator): def __init__(self, device: Optional[torch.device] = None): super().__init__("cpu")
