Revert "Delete unnecessary arg setting (#978)" (#985)

This reverts commit 83c69ecd49.
This commit is contained in:
yzhang93
2023-02-09 16:44:26 -08:00
committed by GitHub
parent c28cc4c919
commit 4ca84ee4ee

View File

@@ -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,