diff --git a/configs/vae_3d/inference/16x256x256.py b/configs/vae_3d/inference/16x256x256.py
index 89591d3..5cda9bd 100644
--- a/configs/vae_3d/inference/16x256x256.py
+++ b/configs/vae_3d/inference/16x256x256.py
@@ -53,6 +53,4 @@ load = None
 
 batch_size = 8
 grad_clip = 1.0
-lr = 1e-4
-
-calc_loss = False
\ No newline at end of file
+lr = 1e-4
\ No newline at end of file
diff --git a/configs/vae_magvit_v2/inference/16x128x128.py b/configs/vae_magvit_v2/inference/16x128x128.py
index 24628ff..b49ac9f 100644
--- a/configs/vae_magvit_v2/inference/16x128x128.py
+++ b/configs/vae_magvit_v2/inference/16x128x128.py
@@ -80,3 +80,5 @@ load = None
 batch_size = 4
 lr = 1e-4
 grad_clip = 1.0
+
+calc_loss = False
\ No newline at end of file