diff --git a/benchmarks/run_all.py b/benchmarks/run_all.py index bef93b4ab31a..779685e452c5 100644 --- a/benchmarks/run_all.py +++ b/benchmarks/run_all.py @@ -48,7 +48,11 @@ def main(): run_command(command.split()) elif file in ["benchmark_sd_img.py", "benchmark_sd_inpainting.py"]: - sdxl_ckpt = "stabilityai/stable-diffusion-xl-base-1.0" + sdxl_ckpt = ( + "stabilityai/stable-diffusion-xl-refiner-1.0" + if "inpainting" not in file + else "stabilityai/stable-diffusion-xl-base-1.0" + ) command = f"python {file} --ckpt {sdxl_ckpt}" run_command(command.split()) diff --git a/src/diffusers/models/__init__.py b/src/diffusers/models/__init__.py index 49ee3ee6af6b..e3794939e25e 100644 --- a/src/diffusers/models/__init__.py +++ b/src/diffusers/models/__init__.py @@ -33,8 +33,8 @@ _import_structure["consistency_decoder_vae"] = ["ConsistencyDecoderVAE"] _import_structure["controlnet"] = ["ControlNetModel"] _import_structure["dual_transformer_2d"] = ["DualTransformer2DModel"] - _import_structure["modeling_utils"] = ["ModelMixin"] _import_structure["embeddings"] = ["ImageProjection"] + _import_structure["modeling_utils"] = ["ModelMixin"] _import_structure["prior_transformer"] = ["PriorTransformer"] _import_structure["t5_film_transformer"] = ["T5FilmDecoder"] _import_structure["transformer_2d"] = ["Transformer2DModel"]