From 4ca84ee4ee328fceebdcdab614dfbb39815cd670 Mon Sep 17 00:00:00 2001 From: yzhang93 Date: Thu, 9 Feb 2023 16:44:26 -0800 Subject: [PATCH] Revert "Delete unnecessary arg setting (#978)" (#985) This reverts commit 83c69ecd495fd284aa8432f2482b148aefba53cf. --- .../src/pipelines/pipeline_shark_stable_diffusion_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/apps/stable_diffusion/src/pipelines/pipeline_shark_stable_diffusion_utils.py b/apps/stable_diffusion/src/pipelines/pipeline_shark_stable_diffusion_utils.py index 1eeeaa9d..854434b3 100644 --- a/apps/stable_diffusion/src/pipelines/pipeline_shark_stable_diffusion_utils.py +++ b/apps/stable_diffusion/src/pipelines/pipeline_shark_stable_diffusion_utils.py @@ -186,6 +186,8 @@ class StableDiffusionPipeline: use_tuned: bool, ): if import_mlir: + # TODO: Delet this when on-the-fly tuning of models work. + use_tuned = False mlir_import = SharkifyStableDiffusionModel( model_id, ckpt_loc,