Method record_kwargs

src/utils/cache.py:30–41

记录层的元数据信息 Args: layer_idx: 层索引 kwargs: 包含路由层信息的字典

(self, layer_idx, kwargs)

        self.value_cache = []

    def record_kwargs(self, layer_idx, kwargs):
        """
        记录层的元数据信息

        Args:
            layer_idx: 层索引
            kwargs: 包含路由层信息的字典
        """
        if layer_idx in self.cache_kwargs:
            self.cache_kwargs[layer_idx].update(kwargs)
        else:
            self.cache_kwargs[layer_idx] = kwargs

    def get_layer_length(self):
        return len(self.cache_kwargs)