From 1d5038280d44a36351cb3aa21aad7a8eff220c94 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Tue, 20 Dec 2022 22:07:06 +0100 Subject: Fix training --- pipelines/util.py | 9 --------- 1 file changed, 9 deletions(-) delete mode 100644 pipelines/util.py (limited to 'pipelines/util.py') 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 @@ -import torch - - -def set_use_memory_efficient_attention_xformers(module: torch.nn.Module, valid: bool) -> None: - if hasattr(module, "set_use_memory_efficient_attention_xformers"): - module.set_use_memory_efficient_attention_xformers(valid) - - for child in module.children(): - set_use_memory_efficient_attention_xformers(child, valid) -- cgit v1.2.3-54-g00ecf