mirror of https://github.com/vladmandic/automatic
6 lines
364 B
Python
6 lines
364 B
Python
import torch
|
|
|
|
from modules.sd_hijack_utils import CondFunc
|
|
|
|
CondFunc('torchsde._brownian.brownian_interval._randn', lambda _, size, dtype, device, seed: torch.randn(size, dtype=dtype, device=torch.device("cpu"), generator=torch.Generator(torch.device("cpu")).manual_seed(int(seed))).to(device), lambda _, size, dtype, device, seed: device.type == 'privateuseone')
|