test cogview1 generation, pass
Showing
- arguments.py 2 additions, 15 deletionsarguments.py
- generation/__init__.py 0 additions, 3 deletionsgeneration/__init__.py
- generation/autoregressive_sampling.py 10 additions, 5 deletionsgeneration/autoregressive_sampling.py
- generation/cuda2d_sampling.py 28 additions, 96 deletionsgeneration/cuda2d_sampling.py
- generation/cuda_2d_sampling.py 13 additions, 12 deletionsgeneration/cuda_2d_sampling.py
- generation/sampling_strategies/__init__.py 2 additions, 1 deletiongeneration/sampling_strategies/__init__.py
- generation/sampling_strategies/base_strategy.py 7 additions, 14 deletionsgeneration/sampling_strategies/base_strategy.py
- generation/sampling_strategies/iterative_entfilter_strategy.py 55 additions, 0 deletions...ation/sampling_strategies/iterative_entfilter_strategy.py
- generation/utils.py 78 additions, 0 deletionsgeneration/utils.py
- inference_cogview.py 97 additions, 0 deletionsinference_cogview.py
- model/cached_autoregressive_model.py 7 additions, 6 deletionsmodel/cached_autoregressive_model.py
- scripts/text2image_cogview.sh 41 additions, 0 deletionsscripts/text2image_cogview.sh
- training/__init__.py 2 additions, 0 deletionstraining/__init__.py
- training/deepspeed_training.py 30 additions, 64 deletionstraining/deepspeed_training.py
- training/model_io.py 4 additions, 2 deletionstraining/model_io.py
Loading
Please register or sign in to comment