class CacheConfig

method

str

缓存算法名称，支持attention_cache和dit_block_cache两种。

blocks_count

int

DiT模型的时空注意力Block个数。

steps_count

int

DiT模型的迭代步数。

step_start

int

开始缓存的迭代步数，默认值为0。

step_interval

int

缓存数据的间隔步数，0 < 'step_interval' < 'steps_count'，默认值为1。

step_end

int

结束缓存的迭代步数，0 < 'step_start' < 'step_end' < 'steps_count'，默认值为10000。

block_start

int

开始缓存的Block层数，0 < 'block_start' < 'blocks_count'，默认值为0。

block_end

int

结束缓存的Block层数，0 < 'block_start' < 'block_end' < 'blocks_count'，默认值为10000。