Skip to content

Add wrappers for Stable Diffusion 3.5 Large model#19584

Open
swha815 wants to merge 1 commit into
pytorch:mainfrom
swha815:MLETORCH-2028_SD3.5Large_wrapper
Open

Add wrappers for Stable Diffusion 3.5 Large model#19584
swha815 wants to merge 1 commit into
pytorch:mainfrom
swha815:MLETORCH-2028_SD3.5Large_wrapper

Conversation

@swha815
Copy link
Copy Markdown

@swha815 swha815 commented May 14, 2026

  • Wrappers for CLIP/T5 text encoders, Transformer, and VAEDecoder contained in SD3.5 Large model.
  • Support for exporting any or all of the above component output generated from dummy input (normal-distributed random values).

Change-Id: I3bc202e2e856a81c6598cf01cdcd8ec158694c25

cc @digantdesai @freddan80 @per @zingo @oscarandersson8218 @mansnils @Sebastian-Larsson @robell @rascani

@swha815 swha815 requested a review from lucylq as a code owner May 14, 2026 10:12
@pytorch-bot
Copy link
Copy Markdown

pytorch-bot Bot commented May 14, 2026

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/19584

Note: Links to docs will display an error until the docs builds have been completed.

❗ 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

❌ 1 Cancelled Job, 3 Unrelated Failures

As of commit e883824 with merge base 4c474af (image):

CANCELLED JOB - The following job was cancelled. Please retry:

FLAKY - The following jobs failed but were likely due to flakiness present on trunk:

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@meta-cla meta-cla Bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label May 14, 2026
@pytorch-bot
Copy link
Copy Markdown

pytorch-bot Bot commented May 14, 2026

Workflows were awaiting approval. CI has now been triggered for the ciflow labels on this PR.

@YufengShi-dudu YufengShi-dudu requested a review from zingo May 14, 2026 10:12
@github-actions
Copy link
Copy Markdown

This PR needs a release notes: label

If your change should be included in the release notes (i.e. would users of this library care about this change?), please use a label starting with release notes:. This helps us keep track and include your important work in the next release notes.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "release notes: none"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.

@zingo zingo added the help wanted Extra attention is needed label May 14, 2026
@zingo
Copy link
Copy Markdown
Collaborator

zingo commented May 14, 2026

Hi @digantdesai adding "Help wanted" to check it's ok to add this to the global example folder?

@zingo zingo added the partner: arm For backend delegation, kernels, demo, etc. from the 3rd-party partner, Arm label May 14, 2026
- Wrappers for CLIP/T5 text encoders, Transformer, and VAEDecoder
  contained in SD3.5 Large model.
- Support for exporting any or all of the above component output
  generated from dummy input (normal-distributed random values).

Signed-off-by: Sangwon Ha <sangwon.ha@arm.com>
Change-Id: I3bc202e2e856a81c6598cf01cdcd8ec158694c25
@swha815 swha815 force-pushed the MLETORCH-2028_SD3.5Large_wrapper branch from b6acece to e883824 Compare May 14, 2026 22:17
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/trunk CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. help wanted Extra attention is needed partner: arm For backend delegation, kernels, demo, etc. from the 3rd-party partner, Arm

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants