aboutsummaryrefslogtreecommitdiff
path: root/modules/hypertile.py
diff options
context:
space:
mode:
authoraria1th <35677394+aria1th@users.noreply.github.com>2023-11-17 09:34:50 +0900
committeraria1th <35677394+aria1th@users.noreply.github.com>2023-11-17 09:34:50 +0900
commitc0725ba2d098a6a78610e7d96ee75f63a32d4e52 (patch)
treed541af2feb2aef254d0265a434d67fc1cc3eecbe /modules/hypertile.py
parentc40be2252ab1c8c289562db208c5ac6618bd8545 (diff)
Fix inverted option issue
I'm pretty sure I was sleepy while implementing this
Diffstat (limited to 'modules/hypertile.py')
-rw-r--r--modules/hypertile.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/hypertile.py b/modules/hypertile.py
index 86acecdc..3a1468c6 100644
--- a/modules/hypertile.py
+++ b/modules/hypertile.py
@@ -337,7 +337,7 @@ def hypertile_context_vae(model:nn.Module, aspect_ratio:float, tile_size:int, op
"""
Returns context manager for VAE
"""
- enabled = not opts.hypertile_split_vae_attn
+ enabled = opts.hypertile_split_vae_attn
swap_size = opts.hypertile_swap_size_vae
max_depth = opts.hypertile_max_depth_vae
tile_size_max = opts.hypertile_max_tile_vae
@@ -355,7 +355,7 @@ def hypertile_context_unet(model:nn.Module, aspect_ratio:float, tile_size:int, o
"""
Returns context manager for U-Net
"""
- enabled = not opts.hypertile_split_unet_attn
+ enabled = opts.hypertile_split_unet_attn
swap_size = opts.hypertile_swap_size_unet
max_depth = opts.hypertile_max_depth_unet
tile_size_max = opts.hypertile_max_tile_unet