diff --git a/predict.py b/predict.py index 71442b4..89e5ece 100644 --- a/predict.py +++ b/predict.py @@ -35,7 +35,7 @@ LLAVA_MODEL_PATH = CKPT_PTH.LLAVA_MODEL_PATH SDXL_CLIP1_PATH = CKPT_PTH.SDXL_CLIP1_PATH SDXL_CLIP2_CACHE = f"{MODEL_CACHE}/models--laion--CLIP-ViT-bigG-14-laion2B-39B-b160k" SDXL_CKPT = f"{MODEL_CACHE}/SDXL_cache/sd_xl_base_1.0_0.9vae.safetensors" -SUPIR_CKPT_F = f"{MODEL_CACHE}/SUPIR_cache/SUsPIR-v0F.ckpt" +SUPIR_CKPT_F = f"{MODEL_CACHE}/SUPIR_cache/SUPIR-v0F.ckpt" SUPIR_CKPT_Q = f"{MODEL_CACHE}/SUPIR_cache/SUPIR-v0Q.ckpt"