Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

SwissArmyTransformer
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Iterations
- Wiki
- Environments
- Terraform modules
- Incidents
- Service Desk

Snippets Groups Projects

b32472a9

Commit b32472a9 authored 3 years ago by Ming Ding

Downloads
- Patches
- Plain Diff

finish testing cogview2 generation

parent 154fc361

No related branches found

No related tags found

No related merge requests found

Changes 15

Expand all Hide whitespace changes

Inline Side-by-side

Showing

generation/autoregressive_sampling.py 2 additions, 1 deletion

generation/autoregressive_sampling.py
generation/cuda2d_sampling.py 8 additions, 6 deletions

generation/cuda2d_sampling.py
generation/sampling_strategies/iterative_entfilter_strategy.py 1 addition, 1 deletion

...ation/sampling_strategies/iterative_entfilter_strategy.py
generation/utils.py 8 additions, 7 deletions

generation/utils.py
inference_cogview.py 1 addition, 0 deletions

inference_cogview.py
inference_cogview2.py 125 additions, 0 deletions

inference_cogview2.py
move_weights.py 10 additions, 0 deletions

move_weights.py
mpu/transformer.py 1 addition, 1 deletion

mpu/transformer.py
scripts/text2image_cogview.sh 1 addition, 1 deletion

scripts/text2image_cogview.sh
scripts/text2image_cogview2.sh 41 additions, 0 deletions

scripts/text2image_cogview2.sh
tokenization/cogview/vqvae_tokenizer.py 9 additions, 8 deletions

tokenization/cogview/vqvae_tokenizer.py
vqvae/__init__.py 3 additions, 1 deletion

vqvae/__init__.py
vqvae/api.py 193 additions, 0 deletions

vqvae/api.py
vqvae/vqvae_diffusion.py 782 additions, 0 deletions

vqvae/vqvae_diffusion.py
vqvae/vqvae_zc.py 17 additions, 10 deletions

vqvae/vqvae_zc.py

with 1202 additions and 36 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment

Strive to be the person your dogs believe you are