From ac3bd4baee194c307dbb3e5d94f338d0bb4c65da Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Mon, 12 Jan 2026 13:36:41 +0530 Subject: [PATCH] ifx --- src/diffusers/schedulers/scheduling_sasolver.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/schedulers/scheduling_sasolver.py b/src/diffusers/schedulers/scheduling_sasolver.py index 2e57833822..dffef54612 100644 --- a/src/diffusers/schedulers/scheduling_sasolver.py +++ b/src/diffusers/schedulers/scheduling_sasolver.py @@ -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,