diff options
Diffstat (limited to 'pipelines')
-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) | ||