diff --git a/configs/opensora-v1-1/train/stage1-2.py b/configs/opensora-v1-1/train/stage1-2.py
index f637b42..944b565 100644
--- a/configs/opensora-v1-1/train/stage1-2.py
+++ b/configs/opensora-v1-1/train/stage1-2.py
@@ -9,11 +9,11 @@ dataset = dict(
 )
 # IMG: 1024 (20%) 512 (30%) 256 (50%) drop (50%)
 bucket_config = {  # 1s/it
-    "144p": {16: (1.0, 8), 32: (1.0, 4), 64: (1.0, 2), 128: (1.0, 1)},
-    "256": {1: (0.5, 32), 16: (0.5, 4), 32: (0.5, 2), 64: (0.5, 1), 128: (0.0, None)},
-    "240p": {16: (0.3, 3), 32: (0.3, 1), 64: (0.0, None)},
-    "512": {1: (0.4, 16)},
-    "1024": {1: (0.3, 4)},
+    "144p": {1: (0.5, 48), 16: (1.0, 6), 32: (1.0, 3), 96: (1.0, 1)},
+    "256": {1: (0.5, 24), 16: (0.5, 3), 48: (0.5, 1), 64: (0.0, None)},
+    "240p": {16: (0.3, 2), 32: (0.3, 1), 64: (0.0, None)},
+    "512": {1: (0.4, 12)},
+    "1024": {1: (0.3, 3)},
 }
 mask_ratios = {
     "mask_no": 0.75,