mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-24 23:28:45 +00:00
remove unnecessary test
This commit is contained in:
parent
8bbcf2048d
commit
0dc3a17fbf
|
@ -1,37 +1,9 @@
|
||||||
import pytest
|
import pytest
|
||||||
from typing import cast
|
from typing import cast
|
||||||
from warnings import warn
|
from warnings import warn
|
||||||
from refiners.foundationals.latent_diffusion.schedulers import Scheduler, DPMSolver, DDIM
|
from refiners.foundationals.latent_diffusion.schedulers import DPMSolver, DDIM
|
||||||
from refiners.fluxion import norm, manual_seed
|
from refiners.fluxion import manual_seed
|
||||||
from torch import linspace, float32, randn, Tensor, allclose, device as Device
|
from torch import randn, Tensor, allclose, device as Device
|
||||||
|
|
||||||
|
|
||||||
def test_scheduler_utils():
|
|
||||||
class DummyScheduler(Scheduler):
|
|
||||||
def __call__(self, x: Tensor, noise: Tensor, step: int) -> Tensor:
|
|
||||||
return Tensor()
|
|
||||||
|
|
||||||
def _generate_timesteps(self) -> Tensor:
|
|
||||||
return Tensor()
|
|
||||||
|
|
||||||
scheduler = DummyScheduler(
|
|
||||||
num_inference_steps=10,
|
|
||||||
num_train_timesteps=20,
|
|
||||||
initial_diffusion_rate=0.1,
|
|
||||||
final_diffusion_rate=0.2,
|
|
||||||
device="cpu",
|
|
||||||
)
|
|
||||||
scale_factors = (
|
|
||||||
1.0
|
|
||||||
- linspace(
|
|
||||||
start=0.1**0.5,
|
|
||||||
end=0.2**0.5,
|
|
||||||
steps=20,
|
|
||||||
dtype=float32,
|
|
||||||
)
|
|
||||||
** 2
|
|
||||||
)
|
|
||||||
assert norm(scheduler.scale_factors - scale_factors) == 0
|
|
||||||
|
|
||||||
|
|
||||||
def test_dpm_solver_diffusers():
|
def test_dpm_solver_diffusers():
|
||||||
|
|
Loading…
Reference in a new issue