hunk.code_window:     """
       def __init__(
               self, img_size=224, patch_size=16, in_chans=3, num_classes=1000, global_pool='token',
               embed_dim=768, depth=12, num_heads=12, mlp_ratio=4., qkv_bias=True, init_values=None,
               drop_rate=0., attn_drop_rate=0., drop_path_rate=0., weight_init='', class_token=True,
               fc_norm=None, embed_layer=PatchEmbed, norm_layer=None, act_layer=None, block_fn=Block):
           """
           Args:
               img_size (int, tuple): input image size
  
hunk.labels: keep keep keep keep keep replace replace keep keep keep
hunk.after_edit:             class_token=True, fc_norm=None, drop_rate=0., attn_drop_rate=0., drop_path_rate=0., weight_init='',
               embed_layer=PatchEmbed, norm_layer=None, act_layer=None, block_fn=Block):
  
hunk.file_path: timm/models/vision_transformer.py
hunk.type: replace
file: model-index==0.1.10
  jinja2==2.11.3
  