hezar.models.image2text.trocr package¶
Submodules¶
- hezar.models.image2text.trocr.trocr_image2text module
TrOCRImage2TextTrOCRImage2Text.compute_loss()TrOCRImage2Text.forward()TrOCRImage2Text.generate()TrOCRImage2Text.image_processorTrOCRImage2Text.is_generativeTrOCRImage2Text.loss_func_nameTrOCRImage2Text.post_process()TrOCRImage2Text.preprocess()TrOCRImage2Text.required_backendsTrOCRImage2Text.tokenizer_name
- hezar.models.image2text.trocr.trocr_image2text_config module
DecoderConfigDecoderConfig.add_cross_attentionDecoderConfig.attention_probs_dropout_probDecoderConfig.bos_token_idDecoderConfig.classifier_dropoutDecoderConfig.eos_token_idDecoderConfig.gradient_checkpointingDecoderConfig.hidden_actDecoderConfig.hidden_dropout_probDecoderConfig.hidden_sizeDecoderConfig.initializer_rangeDecoderConfig.intermediate_sizeDecoderConfig.is_decoderDecoderConfig.layer_norm_epsDecoderConfig.max_position_embeddingsDecoderConfig.num_attention_headsDecoderConfig.num_hidden_layersDecoderConfig.pad_token_idDecoderConfig.position_embedding_typeDecoderConfig.type_vocab_sizeDecoderConfig.use_cacheDecoderConfig.vocab_size
EncoderConfigEncoderConfig.attention_probs_dropout_probEncoderConfig.encoder_strideEncoderConfig.hidden_actEncoderConfig.hidden_dropout_probEncoderConfig.hidden_sizeEncoderConfig.image_sizeEncoderConfig.initializer_rangeEncoderConfig.intermediate_sizeEncoderConfig.layer_norm_epsEncoderConfig.num_attention_headsEncoderConfig.num_channelsEncoderConfig.num_hidden_layersEncoderConfig.patch_sizeEncoderConfig.qkv_bias
GenerationConfigTrOCRImage2TextConfig