+h! S SK r S SKrS SKJr S SKJr S SKJrJr S SK r S SK
J r S SK J
r
SSKJrJr S r " S
S \5 r/ S Q/ S
Q/ SQ/ SQ/ SQS.r\ " S S\5 5 r " S S\5 rg) N) dataclass)Enum)OptionalUnion)validate_hf_hub_args)Self )
BaseOutputPushToHubMixinzscheduler_config.jsonc P \ rS rSrSrSrSrSrSrSr Sr
S r S
r S r
S rS
rSrSrSrSrg)KarrasDiffusionSchedulers" r
N)__name__
__module__ __qualname____firstlineno__
DDIMScheduler
DDPMScheduler
PNDMSchedulerLMSDiscreteSchedulerEulerDiscreteSchedulerHeunDiscreteScheduler EulerAncestralDiscreteSchedulerDPMSolverMultistepScheduler DPMSolverSinglestepSchedulerKDPM2DiscreteScheduler KDPM2AncestralDiscreteSchedulerDEISMultistepSchedulerUniPCMultistepSchedulerDPMSolverSDESchedulerEDMEulerScheduler__static_attributes__r _/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/schedulers/scheduling_utils.pyr
r
" sY MMM &'#"# #$ &(# r2 r
)
iR i i i! i iW | ) {G:-@gfffff@gS@g~jt@gV-?gtV?g7A`?gw/?gK7?g~jt? )
r4 iM i iK i i6 t 5 r ) r8 g(\B@g^I +@g r@gFx?g/$?g(\?gR Q?gʡE?g!rh?r9 ) g @g @K@gʡ/@g r @gˡE@g9v?g1 Zd?gn?g/$?g rh?r9 )StableDiffusionTimestepsStableDiffusionSigmasStableDiffusionXLTimestepsStableDiffusionXLSigmasStableDiffusionVideoSigmasc 8 \ rS rSr% Sr\R \S' Srg)SchedulerOutput= a0
Base class for the output of a scheduler's `step` function.
Args:
prev_sample (`torch.Tensor` of shape `(batch_size, num_channels, height, width)` for images):
Computed sample `(x_{t-1})` of previous timestep. `prev_sample` should be used as next model input in the
denoising loop.
prev_sampler N) r r r r! __doc__torchTensor__annotations__r1 r r2 r3 rC rC = s r2 rC c \ rS rSrSr\r/ rSr\ \
SS\ \ \
\R 4 S\ \
S\4S jj5 5 rSS \ \
\R 4 S
\4S jjr\S 5 r\ S
5 rSrg)SchedulerMixinK a
Base class for all schedulers.
[`SchedulerMixin`] contains common functions shared by all schedulers such as general loading and saving
functionalities.
[`ConfigMixin`] takes care of storing the configuration attributes (like `num_train_timesteps`) that are passed to
the scheduler's `__init__` function, and the attributes can be accessed by `scheduler.config.num_train_timesteps`.
Class attributes:
- **_compatibles** (`List[str]`) -- A list of scheduler classes that are compatible with the parent scheduler
class. Use [`~ConfigMixin.from_config`] to load a different compatible scheduler class (should be overridden
by parent class).
TN pretrained_model_name_or_path subfolderreturnc ` U R " SUUSSS.UD6u pTnU R " U4SU0UD6$ )a
Instantiate a scheduler from a pre-defined JSON configuration file in a local directory or Hub repository.
Parameters:
pretrained_model_name_or_path (`str` or `os.PathLike`, *optional*):
Can be either:
- A string, the *model id* (for example `google/ddpm-celebahq-256`) of a pretrained model hosted on
the Hub.
- A path to a *directory* (for example `./my_model_directory`) containing the scheduler
configuration saved with [`~SchedulerMixin.save_pretrained`].
subfolder (`str`, *optional*):
The subfolder location of a model file within a larger model repository on the Hub or locally.
return_unused_kwargs (`bool`, *optional*, defaults to `False`):
Whether kwargs that are not consumed by the Python class should be returned or not.
cache_dir (`Union[str, os.PathLike]`, *optional*):
Path to a directory where a downloaded pretrained model configuration is cached if the standard cache
is not used.
force_download (`bool`, *optional*, defaults to `False`):
Whether or not to force the (re-)download of the model weights and configuration files, overriding the
cached versions if they exist.
proxies (`Dict[str, str]`, *optional*):
A dictionary of proxy servers to use by protocol or endpoint, for example, `{'http': 'foo.bar:3128',
'http://hostname': 'foo.bar:4012'}`. The proxies are used on each request.
output_loading_info(`bool`, *optional*, defaults to `False`):
Whether or not to also return a dictionary containing missing keys, unexpected keys and error messages.
local_files_only(`bool`, *optional*, defaults to `False`):
Whether to only load local model weights and configuration files or not. If set to `True`, the model
won't be downloaded from the Hub.
token (`str` or *bool*, *optional*):
The token to use as HTTP bearer authorization for remote files. If `True`, the token generated from
`diffusers-cli login` (stored in `~/.huggingface`) is used.
revision (`str`, *optional*, defaults to `"main"`):
The specific model version to use. It can be a branch name, a tag name, a commit id, or any identifier
allowed by Git.
To use private or [gated models](https://huggingface.co/docs/hub/models-gated#gated-models), log-in with `hf
auth login`. You can also activate the special
["offline-mode"](https://huggingface.co/diffusers/installation.html#offline-mode) to use this method in a
firewalled environment.
T)rM rN return_unused_kwargsreturn_commit_hashrQ r ) load_config from_config)clsrM rN rQ kwargsconfig commit_hashs r3 from_pretrained SchedulerMixin.from_pretrained_ sP p '*oo '
*G !% # '
'
# v[r sx ! " 6 " . 0 & !R i"R k"n
j
w"^ w"r2