-
- Downloads
tmp finish naive ar sampling
Showing
- generation/__init__.py 0 additions, 1 deletiongeneration/__init__.py
- generation/autoregressive_sampling.py 99 additions, 0 deletionsgeneration/autoregressive_sampling.py
- generation/cuda2d_sampling.py 157 additions, 0 deletionsgeneration/cuda2d_sampling.py
- generation/local_sampling.py 0 additions, 147 deletionsgeneration/local_sampling.py
- generation/sampling_strategies/__init__.py 1 addition, 0 deletionsgeneration/sampling_strategies/__init__.py
- generation/sampling_strategies/base_strategy.py 46 additions, 0 deletionsgeneration/sampling_strategies/base_strategy.py
- model/cached_autoregressive_model.py 2 additions, 1 deletionmodel/cached_autoregressive_model.py
- model/mixins.py 2 additions, 2 deletionsmodel/mixins.py
- mpu/transformer.py 2 additions, 3 deletionsmpu/transformer.py
- pretrain_cogview2.py 1 addition, 1 deletionpretrain_cogview2.py
- scripts/finetune_into_cogview2.sh 59 additions, 0 deletionsscripts/finetune_into_cogview2.sh
- training/model_io.py 1 addition, 1 deletiontraining/model_io.py
Loading
Please register or sign in to comment