Update documents, add new options information

This commit is contained in:
ArDiouscuros 2022-09-27 08:59:16 +02:00 committed by Lincoln Stein
parent c2e100e6bf
commit cad237b4c8
3 changed files with 13 additions and 0 deletions

View File

@ -98,6 +98,8 @@ overridden on a per-prompt basis (see [List of prompt arguments]
| `--gfpgan_dir` | | `src/gfpgan` | Path to where GFPGAN is installed. |
| `--gfpgan_model_path` | | `experiments/pretrained_models/GFPGANv1.4.pth` | Path to GFPGAN model file, relative to `--gfpgan_dir`. |
| `--device <device>` | `-d<device>` | `torch.cuda.current_device()` | Device to run SD on, e.g. "cuda:0" |
| `--free_gpu_mem` | | `False` | Free GPU memory after sampling, to allow image decoding and saving in low VRAM conditions |
| `--precision` | | `auto` | Set model precision, default is selected by device. Options: auto, float32, float16, autocast |
#### deprecated

View File

@ -116,3 +116,13 @@ branch that contains the pull request:
You will need to go through the install procedure again, but it should be fast because all the
dependencies are already loaded.
---
### **QUESTION**
Image generation crashed with CUDA out of memory error after successful sampling.
### **SOLUTION**
Try to run script with option `--free_gpu_mem` This will free memory before image decoding step.

View File

@ -57,6 +57,7 @@ We thank them for all of their time and hard work.
- [Kyle Schouviller](https://github.com/kyle0654)
- [rabidcopy](https://github.com/rabidcopy)
- [Dominic Letz](https://github.com/dominicletz)
- [Dmitry T.](https://github.com/ArDiouscuros)
## **Original CompVis Authors:**