diff --git a/README.md b/README.md index 76402fe..ad965e2 100644 --- a/README.md +++ b/README.md @@ -27,6 +27,7 @@ conda activate SUPIR pip install --upgrade pip pip install -r requirements.txt + pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 ``` 3. Download Checkpoints @@ -57,7 +58,12 @@ For users who can connect to huggingface, please setting `LLAVA_CLIP_PATH, SDXL_ 4. Edit Custom Path for Checkpoints ``` * [CKPT_PTH.py] --> LLAVA_CLIP_PATH, LLAVA_MODEL_PATH, SDXL_CLIP1_PATH, SDXL_CLIP2_CACHE_DIR - * [options/SUPIR_v0.yaml] --> SDXL_CKPT, SUPIR_CKPT_Q, SUPIR_CKPT_F + * [options/SUPIR_v0.yaml] --> SDXL_CKPT, SUPIR_CKPT_Q, SUPIR_CKPT_F + +5. If running locally, you must edit /options/SUPIR_v0.yaml to point to the local locations of the model files. Example syntax for WSK and Window is in SUPIR_v0.yaml + +6. 5. If running locally, you must also edit CKPT_PTH.py to point to the local locations of the model files. + ``` --- diff --git a/options/SUPIR_v0.yaml b/options/SUPIR_v0.yaml index 99d76e5..2a6ed33 100644 --- a/options/SUPIR_v0.yaml +++ b/options/SUPIR_v0.yaml @@ -148,7 +148,11 @@ model: 'painting, oil painting, illustration, drawing, art, sketch, oil painting, cartoon, CG Style, 3D render, unreal engine, blurring, dirty, messy, worst quality, low quality, frames, watermark, signature, jpeg artifacts, deformed, lowres, over-smooth' - +# +# The 4 locations below need to be modified to point to the local locations of the models. +# Under WSL, for examole, the syntax is: SDXL_CKPT: './opt/data/Juggernaut-XL_v9_RunDiffusionPhoto_v2.safetensors' +# Under Windows, the syntaz is: SDXL_CKPT: A:\SUPIR\opt\data\Juggernaut-XL_v9_RunDiffusionPhoto_v2.safetensors +# SDXL_CKPT: /opt/data/private/AIGC_pretrain/SDXL_cache/sd_xl_base_1.0_0.9vae.safetensors SUPIR_CKPT_F: /opt/data/private/AIGC_pretrain/SUPIR_cache/SUPIR-v0F.ckpt SUPIR_CKPT_Q: /opt/data/private/AIGC_pretrain/SUPIR_cache/SUPIR-v0Q.ckpt diff --git a/requirements.txt b/requirements.txt index 11e74eb..a7a1ee1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -33,7 +33,6 @@ pytorch-lightning==2.1.2 PyYAML==6.0 scipy==1.9.1 tqdm==4.65.0 -triton==2.1.0 urllib3==1.26.15 webdataset==0.2.48 xformers>=0.0.20