Skip to content

Commit 883c85e

Browse files
committed
fix bug (flash pipeline not selected)
1 parent 9c61a46 commit 883c85e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/models/model_31/model_kd31.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def __init__(self, params: KubinParams):
2727

2828
self.params = params
2929

30-
self.use_flash_pipeline = False
30+
self.use_flash_pipeline = self.params("native", "use_kandinsky_flash")
3131
self.t2i_pipe: Kandinsky3T2IPipeline | Kandinsky3T2ILowVRAMPipeline | None = (
3232
None
3333
)
@@ -57,7 +57,7 @@ def prepare_model(self, task):
5757
self.flush(task)
5858

5959
if self.use_flash_pipeline:
60-
k_log(f"running flash K3 pipeline")
60+
k_log(f"preparing flash K3 pipeline")
6161

6262
self.t2i_pipe = get_T2I_Flash_pipeline(
6363
environment=environment,
@@ -76,7 +76,7 @@ def prepare_model(self, task):
7676
movq_path=None,
7777
)
7878
else:
79-
k_log(f"running regular K3 pipeline")
79+
k_log(f"preparing regular K3 pipeline")
8080

8181
self.t2i_pipe = get_T2I_pipeline(
8282
environment=environment,

0 commit comments

Comments
 (0)