Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

some installation fixes #92

Open
wants to merge 7 commits into
base: master
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 7 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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.

```
---

Expand Down
6 changes: 5 additions & 1 deletion options/SUPIR_v0.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
1 change: 0 additions & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down