hezar.models.text_generation.t5 package

Submodules

• hezar.models.text_generation.t5.t5_text_generation module
  • T5TextGeneration
    • T5TextGeneration.compute_loss()
    • T5TextGeneration.forward()
    • T5TextGeneration.generate()
    • T5TextGeneration.is_generative
    • T5TextGeneration.loss_func_name
    • T5TextGeneration.post_process()
    • T5TextGeneration.preprocess()
    • T5TextGeneration.required_backends
• hezar.models.text_generation.t5.t5_text_generation_config module
  • T5TextGenerationConfig
    • T5TextGenerationConfig.d_ff
    • T5TextGenerationConfig.d_kv
    • T5TextGenerationConfig.d_model
    • T5TextGenerationConfig.decoder_start_token_id
    • T5TextGenerationConfig.dropout_rate
    • T5TextGenerationConfig.eos_token_id
    • T5TextGenerationConfig.feed_forward_proj
    • T5TextGenerationConfig.initializer_factor
    • T5TextGenerationConfig.input_prefix
    • T5TextGenerationConfig.is_encoder_decoder
    • T5TextGenerationConfig.layer_norm_epsilon
    • T5TextGenerationConfig.max_length
    • T5TextGenerationConfig.min_length
    • T5TextGenerationConfig.name
    • T5TextGenerationConfig.num_decoder_layers
    • T5TextGenerationConfig.num_heads
    • T5TextGenerationConfig.num_layers
    • T5TextGenerationConfig.pad_token_id
    • T5TextGenerationConfig.relative_attention_max_distance
    • T5TextGenerationConfig.relative_attention_num_buckets
    • T5TextGenerationConfig.tie_word_embeddings
    • T5TextGenerationConfig.use_cache
    • T5TextGenerationConfig.vocab_size

Module contents