diff options
author | Volpeon <git@volpeon.ink> | 2022-12-04 09:24:37 +0100 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2022-12-04 09:24:37 +0100 |
commit | 2825ba2f2030b2fd3e841aad416a4fd28d67615a (patch) | |
tree | 180afc9ae42c7d29b7229423d4db53183e6ad49b /pipelines/util.py | |
parent | Update (diff) | |
download | textual-inversion-diff-2825ba2f2030b2fd3e841aad416a4fd28d67615a.tar.gz textual-inversion-diff-2825ba2f2030b2fd3e841aad416a4fd28d67615a.tar.bz2 textual-inversion-diff-2825ba2f2030b2fd3e841aad416a4fd28d67615a.zip |
Update
Diffstat (limited to 'pipelines/util.py')
-rw-r--r-- | pipelines/util.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/pipelines/util.py b/pipelines/util.py new file mode 100644 index 0000000..661dbee --- /dev/null +++ b/pipelines/util.py | |||
@@ -0,0 +1,9 @@ | |||
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) | ||