Version 2.0.0
NOTE: This needs at least 24GB VRAM for 1x upscale. If you want to upscale more than 1x, you will need more than 24GB of VRAM. 48GB VRAM is recommended.
NOTE: Loading of models on start takes a few minutes, so you can view the log to watch the progress. You will be able to access the port when you see Running on local URL: http://0.0.0.0:3001 in the log.
This template includes:
- Ubuntu 22.04 LTS
- CUDA 11.8
- Python 3.10.12
- SUPIR
- Torch 2.2.0
- xformers 0.0.24
- Jupyter Lab
- runpodctl
- OhMyRunPod
- RunPod File Uploader
- croc
- rclone
Get it here.
Note: You can add an environment variable called NO_GPU_OPTIMIZATION if you are using H100 or A100 GPUs with 80GB of VRAM.