1
0
Fork 0
onnx-web/api/onnx_web/diffusers/stub_scheduler.py

9 lines
246 B
Python

from typing import Any
class StubScheduler:
def step(
self, model_output: Any, timestep: int, sample: Any, return_dict: bool = True
) -> None:
raise NotImplementedError("scheduler not available, try updating diffusers")