#from wan.modules import WanModel

def cache_init(self, num_steps= 50):   
    '''
    Initialization for cache.
    '''
    cache_dic = {}
    cache = {}
    cache[-1]={}
    cache[-1]['cond_stream']={}
    cache[-1]['uncond_stream']={}
    cache_dic['cache_counter'] = 0

    for j in range(self.num_layers):
        cache[-1]['cond_stream'][j] = {}
        cache[-1]['uncond_stream'][j] = {}

    cache_dic['taylor_cache'] = False
    cache_dic['Delta-DiT'] = False


    cache_dic['cache_type'] = 'random'
    cache_dic['fresh_ratio_schedule'] = 'ToCa' 
    cache_dic['fresh_ratio'] = 0.0
    cache_dic['fresh_threshold'] = 1
    cache_dic['force_fresh'] = 'global'

    mode = 'Taylor'

    if mode == 'original':
        cache_dic['cache'] = cache
        cache_dic['force_fresh'] = 'global'
        cache_dic['max_order'] = 0
        cache_dic['first_enhance'] = 3
        
    elif mode == 'ToCa':
        cache_dic['cache_type'] = 'attention'
        cache_dic['cache'] = cache
        cache_dic['fresh_ratio_schedule'] = 'ToCa' 
        cache_dic['fresh_ratio'] = 0.1
        cache_dic['fresh_threshold'] = 5
        cache_dic['force_fresh'] = 'global' 
        cache_dic['soft_fresh_weight'] = 0.0
        cache_dic['max_order'] = 0
        cache_dic['first_enhance'] = 3
    
    elif mode == 'Taylor':
        cache_dic['cache'] = cache
        cache_dic['fresh_threshold'] = 5
        cache_dic['taylor_cache'] = True
        cache_dic['max_order'] = 1
        cache_dic['first_enhance'] = 1

    elif mode == 'Delta':
        cache_dic['cache'] = cache
        cache_dic['fresh_ratio'] = 0.0
        cache_dic['fresh_threshold'] = 3
        cache_dic['Delta-DiT'] = True
        cache_dic['max_order'] = 0
        cache_dic['first_enhance'] = 1

    current = {}
    current['activated_steps'] = [0]
    current['step'] = 0
    current['num_steps'] = num_steps

    return cache_dic, current