summaryrefslogtreecommitdiffstats
path: root/pipelines/util.py
diff options
context:
space:
mode:
Diffstat (limited to 'pipelines/util.py')
-rw-r--r--pipelines/util.py9
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 @@
1import torch
2
3
4def 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)