diff options
| author | Volpeon <git@volpeon.ink> | 2022-12-28 18:08:36 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-12-28 18:08:36 +0100 |
| commit | 83725794618164210a12843381724252fdd82cc2 (patch) | |
| tree | ec29ade9891fe08dd10b5033214fc09237c2cb86 /training/util.py | |
| parent | Improved learning rate finder (diff) | |
| download | textual-inversion-diff-83725794618164210a12843381724252fdd82cc2.tar.gz textual-inversion-diff-83725794618164210a12843381724252fdd82cc2.tar.bz2 textual-inversion-diff-83725794618164210a12843381724252fdd82cc2.zip | |
Integrated updates from diffusers
Diffstat (limited to 'training/util.py')
| -rw-r--r-- | training/util.py | 5 |
1 files changed, 0 insertions, 5 deletions
diff --git a/training/util.py b/training/util.py index a0c15cd..d0f7fcd 100644 --- a/training/util.py +++ b/training/util.py | |||
| @@ -5,11 +5,6 @@ import torch | |||
| 5 | from PIL import Image | 5 | from PIL import Image |
| 6 | 6 | ||
| 7 | 7 | ||
| 8 | def freeze_params(params): | ||
| 9 | for param in params: | ||
| 10 | param.requires_grad = False | ||
| 11 | |||
| 12 | |||
| 13 | def save_args(basepath: Path, args, extra={}): | 8 | def save_args(basepath: Path, args, extra={}): |
| 14 | info = {"args": vars(args)} | 9 | info = {"args": vars(args)} |
| 15 | info["args"].update(extra) | 10 | info["args"].update(extra) |