Skip to content

Commit 4ef88d6

Browse files
Add wrappers for Stable Diffusion 3.5 Large model
- 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
1 parent b32c2c3 commit 4ef88d6

3 files changed

Lines changed: 888 additions & 0 deletions

File tree

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
# Copyright 2026 Arm Limited and/or its affiliates.
2+
#
3+
# This source code is licensed under the BSD-style license found in the
4+
# LICENSE file in the root directory of this source tree.
5+
6+
from .export_sd3_5_large import parse_component, parse_dtype, SD35LargeExporter
7+
from .model import (
8+
MODEL_ID,
9+
SD3CLIPTextEncoderWrapper,
10+
SD3T5TextEncoderWrapper,
11+
SD3TransformerWrapper,
12+
SD3VAEDecoderWrapper,
13+
StableDiffusion3ModelLoader,
14+
StableDiffusionComponent,
15+
)
16+
17+
__all__ = [
18+
"MODEL_ID",
19+
"SD35LargeExporter",
20+
"SD3CLIPTextEncoderWrapper",
21+
"SD3T5TextEncoderWrapper",
22+
"SD3TransformerWrapper",
23+
"SD3VAEDecoderWrapper",
24+
"StableDiffusion3ModelLoader",
25+
"StableDiffusionComponent",
26+
"parse_component",
27+
"parse_dtype",
28+
]

0 commit comments

Comments
 (0)