Merge pull request #4 from THUDM/package
Package
No related branches found
No related tags found
Showing
- .gitignore 6 additions, 1 deletion.gitignore
- CHANGE_LOG.md 35 additions, 0 deletionsCHANGE_LOG.md
- MANIFEST.in 3 additions, 0 deletionsMANIFEST.in
- README.md 21 additions, 0 deletionsREADME.md
- SwissArmyTransformer/__init__.py 4 additions, 0 deletionsSwissArmyTransformer/__init__.py
- SwissArmyTransformer/arguments.py 39 additions, 1 deletionSwissArmyTransformer/arguments.py
- SwissArmyTransformer/data_utils/__init__.py 0 additions, 0 deletionsSwissArmyTransformer/data_utils/__init__.py
- SwissArmyTransformer/data_utils/configure_data.py 1 addition, 1 deletionSwissArmyTransformer/data_utils/configure_data.py
- SwissArmyTransformer/data_utils/datasets.py 1 addition, 6 deletionsSwissArmyTransformer/data_utils/datasets.py
- SwissArmyTransformer/data_utils/samplers.py 0 additions, 0 deletionsSwissArmyTransformer/data_utils/samplers.py
- SwissArmyTransformer/generation/__init__.py 0 additions, 0 deletionsSwissArmyTransformer/generation/__init__.py
- SwissArmyTransformer/generation/autoregressive_sampling.py 53 additions, 14 deletionsSwissArmyTransformer/generation/autoregressive_sampling.py
- SwissArmyTransformer/generation/cuda2d_sampling.py 1 addition, 1 deletionSwissArmyTransformer/generation/cuda2d_sampling.py
- SwissArmyTransformer/generation/magnify.py 0 additions, 0 deletionsSwissArmyTransformer/generation/magnify.py
- SwissArmyTransformer/generation/sampling_strategies/__init__.py 2 additions, 1 deletion...rmyTransformer/generation/sampling_strategies/__init__.py
- SwissArmyTransformer/generation/sampling_strategies/base_strategy.py 79 additions, 0 deletions...ansformer/generation/sampling_strategies/base_strategy.py
- SwissArmyTransformer/generation/sampling_strategies/beam_search_strategy.py 116 additions, 0 deletions...er/generation/sampling_strategies/beam_search_strategy.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 5 additions, 0 deletionsSwissArmyTransformer/model/__init__.py
Loading
Please register or sign in to comment