[CI][Diffusion] Add tiny model builder for Krea2Pipeline (#6421)
Signed-off-by: khairulkabir1661 <khairulkabir1661@users.noreply.github.com> Co-authored-by: khairulkabir1661 <khairulkabir1661@users.noreply.github.com>
This commit is contained in:
1 parent
62ebad2d7c
commit
bcb4ae288f
4 files changed
+85
-11
No files matched your search
@@ -175,4 +175,9 @@ DIFFUSION_TEST_SETTINGS = {
|
||||
[DiffusionAccs.CFG_PARALLEL, DiffusionAccs.TENSOR_PARALLEL, DiffusionAccs.CPU_OFFLOAD],
|
||||
],
|
||||
),
|
||||
"Krea2Pipeline": DiffusionModelTestOpts(
|
||||
model="krea/Krea-2-Turbo",
|
||||
builder=diff_model_builders.tiny_krea2_builder,
|
||||
supported_tasks=[DiffusionTasks.TEXT_TO_IMAGE],
|
||||
),
|
||||
}
|
||||
Reference in new issue
Block a user