o پi@s,ddlmZmZddlZGdddeZdS))ABCabstractmethodNc@s|eZdZUejed<eed<eed<dddZede ddfd d Z edd d Z e dd ejdedBdejfddZ dS) BaseScheduler timestepsordernum_train_timestepsreturnNcOs0gd}|D]}t||std|dqdS)N)rrrz)Subclasses of BaseScheduler must define 'z ' property)hasattrAttributeError)selfargskwargsrequired_attributesattrrh/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/runtime/models/schedulers/base.py__init__s  zBaseScheduler.__init__shiftcCdSNr)r rrrr set_shiftzBaseScheduler.set_shiftcOrrr)r r r rrr set_timestepsrzBaseScheduler.set_timestepssampletimestepcCrrr)r rrrrrscale_model_input!szBaseScheduler.scale_model_input)rNr) __name__ __module__ __qualname__torchTensor__annotations__intrrfloatrrrrrrrr s$    r)abcrrrrrrrrs