mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-10 06:17:55 -05:00
c124b76328
The stable diffusion codebase has been reorganized to make it more modular so that the same script can be used for web as well as cli, instead of duplicating the whole codebase. Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com>
96 lines
2.7 KiB
JSON
96 lines
2.7 KiB
JSON
{
|
|
"unet": {
|
|
"tuned": {
|
|
"fp16": {
|
|
"default_compilation_flags": []
|
|
},
|
|
"fp32": {
|
|
"default_compilation_flags": []
|
|
}
|
|
},
|
|
"untuned": {
|
|
"fp16": {
|
|
"default_compilation_flags": [
|
|
"--iree-flow-enable-padding-linalg-ops",
|
|
"--iree-flow-linalg-ops-padding-size=32"
|
|
],
|
|
"specified_compilation_flags": {
|
|
"cuda": ["--iree-flow-enable-conv-nchw-to-nhwc-transform"],
|
|
"default_device": ["--iree-flow-enable-conv-img2col-transform"]
|
|
}
|
|
},
|
|
"fp32": {
|
|
"default_compilation_flags": [
|
|
"--iree-flow-enable-conv-nchw-to-nhwc-transform",
|
|
"--iree-flow-enable-padding-linalg-ops",
|
|
"--iree-flow-linalg-ops-padding-size=16"
|
|
]
|
|
}
|
|
}
|
|
},
|
|
"vae": {
|
|
"tuned": {
|
|
"fp16": {
|
|
"default_compilation_flags": [
|
|
"--iree-flow-enable-padding-linalg-ops",
|
|
"--iree-flow-linalg-ops-padding-size=32",
|
|
"--iree-flow-enable-conv-img2col-transform"
|
|
]
|
|
},
|
|
"fp32": {
|
|
"default_compilation_flags": [
|
|
"--iree-flow-enable-padding-linalg-ops",
|
|
"--iree-flow-linalg-ops-padding-size=32",
|
|
"--iree-flow-enable-conv-img2col-transform"
|
|
]
|
|
}
|
|
},
|
|
"untuned": {
|
|
"fp16": {
|
|
"default_compilation_flags": [
|
|
"--iree-flow-enable-padding-linalg-ops",
|
|
"--iree-flow-linalg-ops-padding-size=32",
|
|
"--iree-flow-enable-conv-img2col-transform"
|
|
]
|
|
},
|
|
"fp32": {
|
|
"default_compilation_flags": [
|
|
"--iree-flow-enable-conv-nchw-to-nhwc-transform",
|
|
"--iree-flow-enable-padding-linalg-ops",
|
|
"--iree-flow-linalg-ops-padding-size=16"
|
|
]
|
|
}
|
|
}
|
|
},
|
|
"clip": {
|
|
"tuned": {
|
|
"fp16": {
|
|
"default_compilation_flags": [
|
|
"--iree-flow-linalg-ops-padding-size=16",
|
|
"--iree-flow-enable-padding-linalg-ops"
|
|
]
|
|
},
|
|
"fp32": {
|
|
"default_compilation_flags": [
|
|
"--iree-flow-linalg-ops-padding-size=16",
|
|
"--iree-flow-enable-padding-linalg-ops"
|
|
]
|
|
}
|
|
},
|
|
"untuned": {
|
|
"fp16": {
|
|
"default_compilation_flags": [
|
|
"--iree-flow-linalg-ops-padding-size=16",
|
|
"--iree-flow-enable-padding-linalg-ops"
|
|
]
|
|
},
|
|
"fp32": {
|
|
"default_compilation_flags": [
|
|
"--iree-flow-linalg-ops-padding-size=16",
|
|
"--iree-flow-enable-padding-linalg-ops"
|
|
]
|
|
}
|
|
}
|
|
}
|
|
}
|