From 7570e6949aa064420667b1694e1f750af0e6e029 Mon Sep 17 00:00:00 2001 From: saienduri <77521230+saienduri@users.noreply.github.com> Date: Mon, 15 Jul 2024 19:22:06 -0700 Subject: [PATCH] Fix typo for hf model name (#777) Otherwise goes down the wrong pipeline --- models/turbine_models/custom_models/sd_inference/sd_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/turbine_models/custom_models/sd_inference/sd_pipeline.py b/models/turbine_models/custom_models/sd_inference/sd_pipeline.py index bd537f29..277f74cb 100644 --- a/models/turbine_models/custom_models/sd_inference/sd_pipeline.py +++ b/models/turbine_models/custom_models/sd_inference/sd_pipeline.py @@ -189,7 +189,7 @@ def get_sd_model_map(hf_model_name): if name in [ "stabilityai/sdxl-turbo", "stabilityai/stable-diffusion-xl-base-1.0", - "/models/SDXL/official_pytorch/fp16/stable_diffusion_fp16//checkpoint_pipe", + "/models/SDXL/official_pytorch/fp16/stable_diffusion_fp16/checkpoint_pipe", ]: return sdxl_model_map elif "stabilityai/stable-diffusion-3" in name: