Lightning-AI / Lightning-AI/lightning-thunder

Support NeMo StableDiffusion network

Open
#266 3 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

enhancement nemo
Dominant language
Python
Stars
1.5k
Forks
121
PR merge metrics
No merged PRs in 30d

Description

🚀 Feature

NeMo's "MegatronLatentDiffusion" model implements Stable Diffusion

Initial examine:

Found 37 distinct operations, of which 30 (81.1%) are supported

Motivation
Pitch
Work items
  • #174
  • #172
  • #177
  • #267
  • #268
  • #269
  • Test memory use of NeMo model with out-of-place add instead of the in-place variant
  • #455
  • #457
  • #458

cc @tfogal

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start with the initial examine result showing 30 of 37 operations supported, then review work items #174, #172, #177, #267, and #268 along with the completed items. Done means completing the remaining NeMo StableDiffusion support work and testing memory use with out-of-place add.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.