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

bd3a2f91

Commit bd3a2f91 authored 3 years ago by zhuoyiyang

Downloads
- Plain Diff

Merge branch 'main' into object_detection

parents 7ab4d90b 465c146b

No related branches found

No related tags found

Loading

Changes 11

Hide whitespace changes

Inline Side-by-side

Showing

arguments.py 5 additions, 3 deletions

arguments.py
generation/autoregressive_sampling.py 20 additions, 13 deletions

generation/autoregressive_sampling.py
inference_cogview_caps.py 93 additions, 0 deletions

inference_cogview_caps.py
model/base_model.py 3 additions, 1 deletion

model/base_model.py
model/cached_autoregressive_model.py 2 additions, 2 deletions

model/cached_autoregressive_model.py
model/cuda2d_model.py 2 additions, 2 deletions

model/cuda2d_model.py
mpu/transformer.py 63 additions, 24 deletions

mpu/transformer.py
scripts/large_scale_select.sh 52 additions, 0 deletions

scripts/large_scale_select.sh
scripts/text2image_cogview.sh 0 additions, 1 deletion

scripts/text2image_cogview.sh
training/deepspeed_training.py 2 additions, 5 deletions

training/deepspeed_training.py
training/model_io.py 3 additions, 3 deletions

training/model_io.py

with 245 additions and 54 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