+hX,SSKJr SSKJrJrJr SSKrSSKrSSK J r J r SSK J r SSKJrJr SSKJr \R&"\5r\"S S \55r"S S \ \ 5rg) ) dataclass)OptionalTupleUnionN) ConfigMixinregister_to_config)SchedulerMixin) BaseOutputlogging) randn_tensorc`\rSrSr%Sr\R \S'Sr\ \R \S'Sr g)SCMSchedulerOutput!aM Output class for the scheduler's `step` function output. 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. pred_original_sample (`torch.Tensor` of shape `(batch_size, num_channels, height, width)` for images): The predicted denoised sample `(x_{0})` based on the model output from the current timestep. `pred_original_sample` can be used to preview progress or for guidance. prev_sampleNpred_original_sample) __name__ __module__ __qualname____firstlineno____doc__torchTensor__annotations__rr__static_attributes__r]/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/schedulers/scheduling_scm.pyrr!s' 37(5<<07rrcZ\rSrSrSrSr\SS\S\S\ 4Sjj5r \ S5r \ S 5r SS \4S jjrS S \S\R S\\\R$4S\ S\ 4 SjjrSrS!SjrS"S\R,S\ S\R,S\R.S\S\\\44 SjjrSrSrg )# SCMScheduler4a `SCMScheduler` extends the denoising procedure introduced in denoising diffusion probabilistic models (DDPMs) with non-Markovian guidance. This model inherits from [`SchedulerMixin`] and [`ConfigMixin`]. Check the superclass documentation for the generic methods the library implements for all schedulers such as loading and saving. Args: num_train_timesteps (`int`, defaults to 1000): The number of diffusion steps to train the model. prediction_type (`str`, defaults to `trigflow`): Prediction type of the scheduler function. Currently only supports "trigflow". sigma_data (`float`, defaults to 0.5): The standard deviation of the noise added during multi-step inference. num_train_timestepsprediction_type sigma_datacSUlSUl[R"[R "SU5SSS2R 5R[R55Ul SUl SUl g)a Initialize the SCM scheduler. Args: num_train_timesteps (`int`, defaults to 1000): The number of diffusion steps to train the model. prediction_type (`str`, defaults to `trigflow`): Prediction type of the scheduler function. Currently only supports "trigflow". sigma_data (`float`, defaults to 0.5): The standard deviation of the noise added during multi-step inference. g?Nr) init_noise_sigmanum_inference_stepsr from_numpynparangecopyastypeint64 timesteps _step_index _begin_index)selfr#r$r%s r__init__SCMScheduler.__init__Fsh&!$$( ))"))A7J*KDbD*Q*V*V*X*_*_`b`h`h*ij rcUR$N)r1r3s r step_indexSCMScheduler.step_indexbsrcUR$r7r2r8s r begin_indexSCMScheduler.begin_indexfs   rr=cXlg)z Sets the begin index for the scheduler. This function should be run from pipeline before the inference. Args: begin_index (`int`): The begin index for the scheduler. Nr<)r3r=s rset_begin_indexSCMScheduler.set_begin_indexks (rNr)r0device max_timestepsintermediate_timestepsc pXRR:a=[SUSURRSURRS35eUb[U5US-:wa [S5eUbUb [S5eUcUc [S 5eUbUS :wa [S 5eXlUb[ U[ 5(a([R"X#S 9R5Ul O[ U[R5(a%URU5R5Ul Or[S [U535eUb+[R"XES/US 9R5Ul O-[R"USUS-US 9R5Ul SUlSUlg)a] Sets the discrete timesteps used for the diffusion chain (to be run before inference). Args: num_inference_steps (`int`): The number of diffusion steps used when generating samples with a pre-trained model. timesteps (`torch.Tensor`, *optional*): Custom timesteps to use for the denoising process. max_timesteps (`float`, defaults to 1.57080): The maximum timestep value used in the SCM scheduler. intermediate_timesteps (`float`, *optional*, defaults to 1.3): The intermediate timestep value used in SCM scheduler (only used when num_inference_steps=2). z`num_inference_steps`: z6 cannot be larger than `self.config.train_timesteps`: zG as the unet model trained with this scheduler can only handle maximal z timesteps.Nr"zWIf providing custom timesteps, `timesteps` must be of length `num_inference_steps + 1`.zFIf providing custom timesteps, `max_timesteps` should not be provided.z5Should provide either `timesteps` or `max_timesteps`.rzNIntermediate timesteps for SCM is not supported when num_inference_steps != 2.)rBzUnsupported timesteps type: r)configr# ValueErrorlenr) isinstancelistrtensorfloatr0rtotypelinspacer1r2)r3r)r0rBrCrDs r set_timestepsSCMScheduler.set_timestepsus* !@!@ @)*=)>?KK3345 KK;;ef f  !6TU U ! -2E2Jmn n#6  )T**!&i!G!M!M!OIu||44!*f!5!;!;!= #?Y?P!QRR # /"\\=RS*T]cdjjlDN#^^M1>QTU>U^dekkmDN rcURc[[U[R5(a%UR UR R 5nURU5UlgURUlgr7) r=rIrrrMr0rBindex_for_timestepr1r2)r3timesteps r_init_step_indexSCMScheduler._init_step_indexsX    #(ELL11#;;t~~'<'<=#66x@D #00D rcUc URnX!:HR5n[U5S:aSOSnX4R5$)Nr"r)r0nonzerorHitem)r3rTschedule_timestepsindicesposs rrSSCMScheduler.index_for_timestepsH  %!% %1::< wN=OPQ Q t~~  "\// 8K8KW`a++(()  ))A,0599Q<%3GGK!K A) )!kXXrc.URR$r7)rFr#r8s r__len__SCMScheduler.__len__s{{...r)r2r1r(r)r0)irdg?)r)NNgH.!?g?r7)NT)rrrrrorderr intstrrLr4propertyr9r=r@rrrrBrPrUrS FloatTensor Generatorboolrrrmrprrrrr r 4sJ  E$() ! !! !!6  !!(3(#'+/&(+ 8! 8!<<8!c5<<'( 8!  8! !& 8!v1 #&&* >Y''>Y>Y!! >Y ?? >Y  >Y !5( )>Y@/rr ) dataclassesrtypingrrrnumpyr+rconfiguration_utilsrr schedulers.scheduling_utilsr utilsr r utils.torch_utilsr get_loggerrloggerrr rrrrs_$")) A8',   H % 88 8"T/>;T/r