diff options
| author | Volpeon <git@volpeon.ink> | 2022-12-20 22:07:06 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-12-20 22:07:06 +0100 |
| commit | 1d5038280d44a36351cb3aa21aad7a8eff220c94 (patch) | |
| tree | 55ae4c0a660f5218c072d33f2896024b47c05c6b /pipelines/util.py | |
| parent | Dependency cleanup/upgrades (diff) | |
| download | textual-inversion-diff-1d5038280d44a36351cb3aa21aad7a8eff220c94.tar.gz textual-inversion-diff-1d5038280d44a36351cb3aa21aad7a8eff220c94.tar.bz2 textual-inversion-diff-1d5038280d44a36351cb3aa21aad7a8eff220c94.zip | |
Fix training
Diffstat (limited to 'pipelines/util.py')
| -rw-r--r-- | pipelines/util.py | 9 |
1 files changed, 0 insertions, 9 deletions
diff --git a/pipelines/util.py b/pipelines/util.py deleted file mode 100644 index 661dbee..0000000 --- a/pipelines/util.py +++ /dev/null | |||
| @@ -1,9 +0,0 @@ | |||
| 1 | import torch | ||
| 2 | |||
| 3 | |||
| 4 | def set_use_memory_efficient_attention_xformers(module: torch.nn.Module, valid: bool) -> None: | ||
| 5 | if hasattr(module, "set_use_memory_efficient_attention_xformers"): | ||
| 6 | module.set_use_memory_efficient_attention_xformers(valid) | ||
| 7 | |||
| 8 | for child in module.children(): | ||
| 9 | set_use_memory_efficient_attention_xformers(child, valid) | ||