o پi@sHddlmZmZddlmZeGdddeZeGdddeZdS)) dataclassfield) CacheParamsc@s<eZdZUdZeed<dZeed<ee dZ e eed<dS)TeaCacheParamsteacache cache_typeteacache_threshdefault_factory coefficientsN) __name__ __module__ __qualname__rstr__annotations__r floatrlistr rra/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/configs/sample/teacache.pyr s  rc@seZdZUdZeed<dZeed<dZe ed<e e dZ e eed<e e dZ e eed <ed e efd d Zed efd dZded efddZdS)WanTeaCacheParamsrrrr T use_ret_stepsr ret_steps_coeffsnon_ret_steps_coeffsreturncCs|jr|jS|jS)N)rrrselfrrrr szWanTeaCacheParams.coefficientscCs|jrdSdS)N rrrrr ret_steps szWanTeaCacheParams.ret_stepsnum_inference_stepscCs|jr|dS|ddS)Nrr)rr!rrrget_cutoff_steps's z"WanTeaCacheParams.get_cutoff_stepsN)r rrrrrr rrboolrrrrpropertyr intr r"rrrrrs   rN) dataclassesrr4sglang.multimodal_gen.configs.sample.sampling_paramsrrrrrrrs