Skip to content

Commit 4ca84ee

Browse files
authored
Revert "Delete unnecessary arg setting (huggingface#978)" (huggingface#985)
This reverts commit 83c69ec.
1 parent c28cc4c commit 4ca84ee

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

apps/stable_diffusion/src/pipelines/pipeline_shark_stable_diffusion_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -186,6 +186,8 @@ def from_pretrained(
186186
use_tuned: bool,
187187
):
188188
if import_mlir:
189+
# TODO: Delet this when on-the-fly tuning of models work.
190+
use_tuned = False
189191
mlir_import = SharkifyStableDiffusionModel(
190192
model_id,
191193
ckpt_loc,

0 commit comments

Comments
 (0)