diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-30 07:11:30 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-30 07:11:30 +0000 |
| commit | 4df63d2d197f26181758b5108f003f225fe84874 (patch) | |
| tree | b7421906e69147a1b76560cd823640d784eee2cc /modules/sd_samplers_common.py | |
| parent | 274474105a5166a985a47508ffd0695db41623a5 (diff) | |
| download | stable-diffusion-webui-gfx803-4df63d2d197f26181758b5108f003f225fe84874.tar.gz stable-diffusion-webui-gfx803-4df63d2d197f26181758b5108f003f225fe84874.tar.bz2 stable-diffusion-webui-gfx803-4df63d2d197f26181758b5108f003f225fe84874.zip | |
split samplers into one more files for k-diffusion
Diffstat (limited to 'modules/sd_samplers_common.py')
| -rw-r--r-- | modules/sd_samplers_common.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/sd_samplers_common.py b/modules/sd_samplers_common.py index 5b06e341..3c03d442 100644 --- a/modules/sd_samplers_common.py +++ b/modules/sd_samplers_common.py @@ -1,4 +1,4 @@ -from collections import namedtuple, deque
+from collections import namedtuple
import numpy as np
import torch
from PIL import Image
@@ -64,6 +64,7 @@ class InterruptedException(BaseException): # MPS fix for randn in torchsde
+# XXX move this to separate file for MPS
def torchsde_randn(size, dtype, device, seed):
if device.type == 'mps':
generator = torch.Generator(devices.cpu).manual_seed(int(seed))
|