-
- Downloads
Merge pull request #6 from THUDM/finer-attn-hooks
Finer attn hooks
No related branches found
No related tags found
Showing
- SwissArmyTransformer/arguments.py 25 additions, 17 deletionsSwissArmyTransformer/arguments.py
- SwissArmyTransformer/data_utils/configure_data.py 12 additions, 13 deletionsSwissArmyTransformer/data_utils/configure_data.py
- SwissArmyTransformer/data_utils/datasets.py 15 additions, 0 deletionsSwissArmyTransformer/data_utils/datasets.py
- SwissArmyTransformer/generation/autoregressive_sampling.py 6 additions, 3 deletionsSwissArmyTransformer/generation/autoregressive_sampling.py
- SwissArmyTransformer/model/__init__.py 2 additions, 1 deletionSwissArmyTransformer/model/__init__.py
- SwissArmyTransformer/model/base_model.py 46 additions, 16 deletionsSwissArmyTransformer/model/base_model.py
- SwissArmyTransformer/model/cached_autoregressive_model.py 20 additions, 32 deletionsSwissArmyTransformer/model/cached_autoregressive_model.py
- SwissArmyTransformer/model/common_layers.py 0 additions, 91 deletionsSwissArmyTransformer/model/common_layers.py
- SwissArmyTransformer/model/cuda2d_model.py 1 addition, 1 deletionSwissArmyTransformer/model/cuda2d_model.py
- SwissArmyTransformer/model/encoder_decoder_model.py 53 additions, 101 deletionsSwissArmyTransformer/model/encoder_decoder_model.py
- SwissArmyTransformer/model/finetune/__init__.py 2 additions, 0 deletionsSwissArmyTransformer/model/finetune/__init__.py
- SwissArmyTransformer/model/finetune/mlp_head.py 36 additions, 0 deletionsSwissArmyTransformer/model/finetune/mlp_head.py
- SwissArmyTransformer/model/finetune/prompt_tuning.py 45 additions, 0 deletionsSwissArmyTransformer/model/finetune/prompt_tuning.py
- SwissArmyTransformer/model/mixins.py 21 additions, 17 deletionsSwissArmyTransformer/model/mixins.py
- SwissArmyTransformer/model/t5_model.py 287 additions, 0 deletionsSwissArmyTransformer/model/t5_model.py
- SwissArmyTransformer/mpu/layers.py 7 additions, 7 deletionsSwissArmyTransformer/mpu/layers.py
- SwissArmyTransformer/mpu/transformer.py 350 additions, 110 deletionsSwissArmyTransformer/mpu/transformer.py
- SwissArmyTransformer/mpu/utils.py 2 additions, 2 deletionsSwissArmyTransformer/mpu/utils.py
- SwissArmyTransformer/tokenization/__init__.py 7 additions, 2 deletionsSwissArmyTransformer/tokenization/__init__.py
- SwissArmyTransformer/tokenization/cogview/sp_tokenizer.py 9 additions, 2 deletionsSwissArmyTransformer/tokenization/cogview/sp_tokenizer.py
Loading
Please register or sign in to comment