fix bug (flash pipeline not selected)

seruva19 · May 25, 2024 · 883c85e · 883c85e
1 parent 9c61a46
commit 883c85e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/models/model_31/model_kd31.py b/src/models/model_31/model_kd31.py
@@ -27,7 +27,7 @@ def __init__(self, params: KubinParams):
 
         self.params = params
 
-        self.use_flash_pipeline = False
+        self.use_flash_pipeline = self.params("native", "use_kandinsky_flash")
         self.t2i_pipe: Kandinsky3T2IPipeline | Kandinsky3T2ILowVRAMPipeline | None = (
             None
         )
@@ -57,7 +57,7 @@ def prepare_model(self, task):
                 self.flush(task)
 
                 if self.use_flash_pipeline:
-                    k_log(f"running flash K3 pipeline")
+                    k_log(f"preparing flash K3 pipeline")
 
                     self.t2i_pipe = get_T2I_Flash_pipeline(
                         environment=environment,
@@ -76,7 +76,7 @@ def prepare_model(self, task):
                         movq_path=None,
                     )
                 else:
-                    k_log(f"running regular K3 pipeline")
+                    k_log(f"preparing regular K3 pipeline")
 
                     self.t2i_pipe = get_T2I_pipeline(
                         environment=environment,