-
- Downloads
Merge branch 'package' of github.com:THUDM/SwissArmyTransformer into package
Conflicts: SwissArmyTransformer/mpu/transformer.py SwissArmyTransformer/training/deepspeed_training.py
No related branches found
No related tags found
Showing
- CHANGE_LOG.md 4 additions, 1 deletionCHANGE_LOG.md
- README.md 6 additions, 1 deletionREADME.md
- SwissArmyTransformer/generation/autoregressive_sampling.py 1 addition, 1 deletionSwissArmyTransformer/generation/autoregressive_sampling.py
- SwissArmyTransformer/generation/cuda2d_sampling.py 1 addition, 1 deletionSwissArmyTransformer/generation/cuda2d_sampling.py
- SwissArmyTransformer/generation/sampling_strategies/base_strategy.py 1 addition, 1 deletion...ansformer/generation/sampling_strategies/base_strategy.py
- SwissArmyTransformer/generation/sampling_strategies/beam_search_strategy.py 1 addition, 1 deletion...er/generation/sampling_strategies/beam_search_strategy.py
- SwissArmyTransformer/generation/sampling_strategies/beam_search_strategy_old.py 0 additions, 467 deletions...eneration/sampling_strategies/beam_search_strategy_old.py
- SwissArmyTransformer/generation/sampling_strategies/iterative_entfilter_strategy.py 1 addition, 1 deletion...ation/sampling_strategies/iterative_entfilter_strategy.py
- SwissArmyTransformer/generation/utils.py 1 addition, 1 deletionSwissArmyTransformer/generation/utils.py
- SwissArmyTransformer/model/__init__.py 1 addition, 0 deletionsSwissArmyTransformer/model/__init__.py
- SwissArmyTransformer/model/base_model.py 1 addition, 1 deletionSwissArmyTransformer/model/base_model.py
- SwissArmyTransformer/model/cached_autoregressive_model.py 1 addition, 1 deletionSwissArmyTransformer/model/cached_autoregressive_model.py
- SwissArmyTransformer/model/common_layers.py 91 additions, 0 deletionsSwissArmyTransformer/model/common_layers.py
- SwissArmyTransformer/model/cuda2d_model.py 1 addition, 1 deletionSwissArmyTransformer/model/cuda2d_model.py
- SwissArmyTransformer/model/encoder_decoder_model.py 144 additions, 0 deletionsSwissArmyTransformer/model/encoder_decoder_model.py
- SwissArmyTransformer/model/mixins.py 1 addition, 1 deletionSwissArmyTransformer/model/mixins.py
- SwissArmyTransformer/mpu/transformer.py 7 additions, 4 deletionsSwissArmyTransformer/mpu/transformer.py
- SwissArmyTransformer/tokenization/__init__.py 4 additions, 3 deletionsSwissArmyTransformer/tokenization/__init__.py
- SwissArmyTransformer/tokenization/cogview/__init__.py 1 addition, 1 deletionSwissArmyTransformer/tokenization/cogview/__init__.py
- SwissArmyTransformer/training/deepspeed_training.py 4 additions, 3 deletionsSwissArmyTransformer/training/deepspeed_training.py
Loading
Please register or sign in to comment