mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-06 10:31:17 -08:00
Code Issues Projects Releases Packages Wiki Activity

change import statements for #14478

Browse source
This commit is contained in:
AUTOMATIC1111 2023-12-31 22:38:30 +03:00
parent be5f1acc8f
commit a70dfb64a8
7 changed files with 14 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

4
test/test_torch_utils.py
View file

@ -3,7 +3,7 @@ import types
import pytest
import torch
from modules.torch_utils import get_param
from modules import torch_utils
@pytest.mark.parametrize("wrapped", [True, False])
@ -14,6 +14,6 @@ def test_get_param(wrapped):
if wrapped:
# more or less how spandrel wraps a thing
mod = types.SimpleNamespace(model=mod)
p = get_param(mod)
p = torch_utils.get_param(mod)
assert p.dtype == torch.float16
assert p.device == cpu