cCiەDSrSSKrSSKJr SSKJr SSKJrJ r SSK r SSK r SSK J r SSKJr SS KJr SS KJr SS KJrJr SS KJrJr SS KJr SSKJrJr SSKJ r J!r!J"r"J#r# SSK$J%r%J&r& SSK'J(r(J)r) SSK*J+r+ \#RX"\-5r.\\""SS9"SS\ 555r/\\""SS9"SS\ 555r0Sr1"SS\ Rd5r3"SS\ Rd5r4SCS \ RdS!\ RjS"\ RjS#\ RjS$\ \ RjS%\6S&\64S'jjr7"S(S)\ Rd5r8"S*S+\ Rd5r9"S,S-\ Rd5r:"S.S/\ Rd5r;"S0S1\ Rd5r<"S2S3\5r="S4S5\ Rd5r>\""S6S7\55r?\""S8S9\?55r@"S:S;\ Rd5rA\""S\?55rB\""S?S9"S@SA\?55rC/SBQrDg)Dz,PyTorch VideoMAE (masked autoencoder) model.N)deepcopy) dataclass)CallableOptional)nn)MSELoss)ACT2FN)GradientCheckpointingLayer)BaseModelOutputImageClassifierOutput)ALL_ATTENTION_FUNCTIONSPreTrainedModel)Unpack) find_pruneable_heads_and_indicesprune_linear_layer) ModelOutputTransformersKwargsauto_docstringlogging)IMAGENET_DEFAULT_MEANIMAGENET_DEFAULT_STD)can_return_tuplecheck_model_inputs)VideoMAEConfigz[ Class for VideoMAEDecoder's outputs, with potential hidden states and attentions. ) custom_introc\rSrSr%SrSr\\R\ S'Sr \\ \R\ S'Sr \\ \R\ S'Sr g)VideoMAEDecoderOutput*zx logits (`torch.FloatTensor` of shape `(batch_size, patch_size ** 2 * num_channels)`): Pixel reconstruction logits. Nlogits hidden_states attentions)__name__ __module__ __qualname____firstlineno____doc__r!rtorch FloatTensor__annotations__r"tupler#__static_attributes__r$h/home/james-whalen/.local/lib/python3.13/site-packages/transformers/models/videomae/modeling_videomae.pyrr*sR +/FHU&& '.8