Skip to content

Commit 596499a

Browse files
authored
Disable tuned configs on all inpainting models (huggingface#1086)
1 parent 2a2e460 commit 596499a

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

apps/stable_diffusion/src/utils/utils.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -244,11 +244,7 @@ def set_init_device_flags():
244244
base_model_id = fetch_and_update_base_model_id(args.ckpt_loc)
245245

246246
if (
247-
args.hf_model_id
248-
in [
249-
"runwayml/stable-diffusion-inpainting",
250-
"stabilityai/stable-diffusion-2-inpainting",
251-
]
247+
"inpainting" in args.hf_model_id
252248
or args.precision != "fp16"
253249
or args.height != 512
254250
or args.width != 512

0 commit comments

Comments
 (0)