src/transformers/models/cvt/modeling_tf_cvt.py/TFCvtEncoder/__init__
class TFCvtEncoder:    def __init__(self, config: CvtConfig, **kwargs):
        super().__init__(**kwargs)
        self.config = config
        self.stages = [
            TFCvtStage(config, stage_idx, name=f"stages.{stage_idx}") for stage_idx in range(len(config.depth))
        ]