Fix typo for hf model name (#777)

Otherwise goes down the wrong pipeline
nod-ai · Jul 16, 2024 · 7570e69 · 7570e69
1 parent 5ee0b20
commit 7570e69
Showing 1 changed file with 1 addition and 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
@@ -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: