Models
Models copied to clipboard
[WIP]开源之夏:MegEngine 补充跨模态模型的实现——DALL-E
参考实现:DALLE-pytorch DALL-E
- [x] OpenaiDicreteVAE编写
- [x] OpenaiDicreteVAE与训练权重转换
- [x] OpenaiDicreteVAE对齐
- [x] Attention编写
- [x] Attention对齐
- [x] SparseConvolutionAttention编写
- [x] SparseConvolutionAttention对齐
- [x] SparseAxialAttention编写
- [x] SparseAxialAttention对齐
- [x] Transformer编写
- [ ] Transformer对齐
- [x] DALLE编写
- [ ] DALLE对齐
- [ ] 其他对齐
- [ ] 推理结果
- [ ] 文档
可能:
- [ ] 添加其他vae模型
- [ ] 可视化SparseAttention的mask
- [ ] 训练代码