diff --git a/_typos.toml b/_typos.toml
index 362ba8a6..bf0292e5 100644
--- a/_typos.toml
+++ b/_typos.toml
@@ -31,6 +31,7 @@ wn="wn"
 hime="hime"
 OT="OT"
 byt="byt"
+tak="tak"
 
 # [files]
 # # Extend the default list of files to check
diff --git a/hunyuan_image_train_network.py b/hunyuan_image_train_network.py
index 291d5132..40c1f2fe 100644
--- a/hunyuan_image_train_network.py
+++ b/hunyuan_image_train_network.py
@@ -394,7 +394,7 @@ class HunyuanImageNetworkTrainer(train_network.NetworkTrainer):
         if self.is_swapping_blocks:
             # Swap blocks between CPU and GPU to reduce memory usage, in forward and backward passes.
             logger.info(f"enable block swap: blocks_to_swap={args.blocks_to_swap}")
-            model.enable_block_swap(args.blocks_to_swap, accelerator.device)
+            model.enable_block_swap(args.blocks_to_swap, accelerator.device, supports_backward=True)
 
         return model, text_encoders