Implementing Mosaic Diffusion into Patch-Diffusion

オープン
#121 コメント 0 件 リアクション 1 件 担当者 0 名 GitHub で見る

まだ誰も着手していません。

評価

難易度
5/5
見積もり時間
1週間以上
初心者へのやさしさ
25/100
issue の種類
機能追加
明瞭さ
説明が足りない
活発さ
停滞
技術スタック
python

調査の方向性

Begin by comparing the Mosaic Diffusion and Patch-Diffusion training scripts, since no specific files or tests are named. Trace the implementations of xFormer/FlashAttention, latent precomputation, low-precision normalization, FSDP, and scheduled EMA; done would require agreeing on scope and integrating the applicable features across the training workflows.

索引モデルが issue の本文から書いたものです。

説明

Patch Diffusion can x2 training speed even on 256x256 ImageNet. If this works out between Mosaic Diffusion and Patch-Diffusion, that is potentially x10 cumulative boost. The issue is both have different training script so I'm thinking of copy+paste the features of Mosaic into Patch-Diffusion. Right now, I only ask where to find relevant code for

  1. xFormer+FlashAttention - I'll be trying to swap FlashAttention-1 for FlashAttention-2
  2. Precomputing latent
  3. Low Precision LayerNorm and GroupNorm
  4. FSDP
  5. Scheduled EMA
主要言語
Python
スター
722
フォーク
75
PR マージ指標
30日以内にマージされた PR はありません

環境構築

  • Dockerfile または Docker Compose ファイルあり
  • プルリクエストのテンプレートなし
  • コントリビューションガイドなし

はじめの一歩

  1. issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
  2. 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
  3. リポジトリをフォークし、ブランチを切って変更します。
  4. issue 番号を参照したプルリクエストを送ります。

mosaicml/diffusion のほかの issue

mosaicml/diffusion の issue をすべて見る

似ている issue

Python の issue をもっと見る

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。