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) | ||
