AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

ifx

Browse Source
This commit is contained in:
sayakpaul
2026-01-12 13:36:41 +05:30
parent e47af9bada
commit ac3bd4baee
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/diffusers/schedulers/scheduling_sasolver.py
View File

@@ -16,7 +16,7 @@
# The codebase is modified based on https://github.com/huggingface/diffusers/blob/main/src/diffusers/schedulers/scheduling_dpmsolver_multistep.py
import math
from typing import Literal
from typing import Callable, Literal
import numpy as np
import torch
@@ -166,7 +166,7 @@ class SASolverScheduler(SchedulerMixin, ConfigMixin):
predictor_order: int = 2,
corrector_order: int = 2,
prediction_type: str = "epsilon",
tau_func: callable | None = None,
tau_func: Callable | None = None,
thresholding: bool = False,
dynamic_thresholding_ratio: float = 0.995,
sample_max_value: float = 1.0,