czk32611

Results 2 issues of czk32611

I ran the training code with two GPUs and got error message `Parameters which did not receive grad for rank 0 down_blocks.2.attentions.1.transformer_blocks.0.attn_temp.to_out.0.weight, down_blocks.2.attentions.1.transformer_blocks.0.attn_temp.to_v.weight, down_blocks.2.attentions.1.transformer_blocks.0.attn_temp.to_k.weight, down_blocks.2.attentions.1.transformer_blocks.0.attn_temp.to_q.weight, xxx`. I double checked and...

### Feature request / 功能建议 In the paper, the authors gave an example for caption upsampler for infererence. It would be great if the authors could provide the exact codes,...