mirror of
https://github.com/brycedrennan/imaginAIry
synced 2024-10-31 03:20:40 +00:00
7af1ab66ca
As documented here: https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/4558#issuecomment-1310387114 - make image logs more consistent - note which step the progress images belong to in the filename
21 lines
627 B
Python
21 lines
627 B
Python
import pytest
|
|
import torch
|
|
|
|
from imaginairy.utils import get_device
|
|
|
|
|
|
@pytest.mark.skipif("mps" not in get_device(), reason="MPS only bug")
|
|
@pytest.mark.xfail(reason="MPS only bug")
|
|
def test_sigma_bug():
|
|
# https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/4558#issuecomment-1310387114
|
|
def t_fn_a(sigma):
|
|
return sigma.to(get_device()).log().neg()
|
|
|
|
def t_fn_b(sigma):
|
|
return sigma.to("cpu").log().neg().to(get_device())
|
|
|
|
sigmas = torch.tensor([0.1, 0.2, 0.3, 0.4, 0.5], device=get_device())
|
|
|
|
for i in range(sigmas.size()[0]):
|
|
assert t_fn_a(sigmas[i]) == t_fn_b(sigmas[i])
|