diff --git a/configs/repa_improved_ddt_xlen22de6_256.yaml b/configs/repa_improved_ddt_xlen22de6_256.yaml
index c8ce2e7..a853d4f 100644
--- a/configs/repa_improved_ddt_xlen22de6_256.yaml
+++ b/configs/repa_improved_ddt_xlen22de6_256.yaml
@@ -2,10 +2,10 @@
 seed_everything: true
 tags:
   exp: &exp repa_flatten_condit22_dit6_fixt_xl
-torch_hub_dir: /mnt/bn/wangshuai6/torch_hub
+torch_hub_dir: null
 huggingface_cache_dir: null
 trainer:
-  default_root_dir: /mnt/bn/wangshuai6/universal_flow_workdirs
+  default_root_dir: universal_flow_workdirs
   accelerator: auto
   strategy: auto
   devices: auto
@@ -40,7 +40,7 @@ model:
     class_path: src.models.vae.LatentVAE
     init_args:
       precompute: true
-      weight_path: /mnt/bn/wangshuai6/models/sd-vae-ft-ema/
+      weight_path: stabilityai/sd-vae-ft-ema
   denoiser:
     class_path: src.models.denoiser.decoupled_improved_dit.DDT
     init_args:
diff --git a/configs/repa_improved_ddt_xlen22de6_512.yaml b/configs/repa_improved_ddt_xlen22de6_512.yaml
index 5c6b665..0e62979 100644
--- a/configs/repa_improved_ddt_xlen22de6_512.yaml
+++ b/configs/repa_improved_ddt_xlen22de6_512.yaml
@@ -2,10 +2,10 @@
 seed_everything: true
 tags:
   exp: &exp res512_fromscratch_repa_flatten_condit22_dit6_fixt_xl
-torch_hub_dir: /mnt/bn/wangshuai6/torch_hub
+torch_hub_dir: null
 huggingface_cache_dir: null
 trainer:
-  default_root_dir: /mnt/bn/wangshuai6/universal_flow_workdirs
+  default_root_dir: universal_flow_workdirs
   accelerator: auto
   strategy: auto
   devices: auto
@@ -40,7 +40,7 @@ model:
     class_path: src.models.vae.LatentVAE
     init_args:
       precompute: true
-      weight_path: /mnt/bn/wangshuai6/models/sd-vae-ft-ema/
+      weight_path: stabilityai/sd-vae-ft-ema
   denoiser:
     class_path: src.models.denoiser.decoupled_improved_dit.DDT
     init_args:
diff --git a/configs/repa_improved_dit_large.yaml b/configs/repa_improved_dit_large.yaml
index a6a51b9..905c892 100644
--- a/configs/repa_improved_dit_large.yaml
+++ b/configs/repa_improved_dit_large.yaml
@@ -2,10 +2,10 @@
 seed_everything: true
 tags:
   exp: &exp repa_improved_dit_large
-torch_hub_dir: /mnt/bn/wangshuai6/torch_hub
+torch_hub_dir: null
 huggingface_cache_dir: null
 trainer:
-  default_root_dir: /mnt/bn/wangshuai6/universal_flow_workdirs
+  default_root_dir: universal_flow_workdirs
   accelerator: auto
   strategy: auto
   devices: auto
@@ -40,7 +40,7 @@ model:
     class_path: src.models.vae.LatentVAE
     init_args:
       precompute: true
-      weight_path: /mnt/bn/wangshuai6/models/sd-vae-ft-ema/
+      weight_path: stabilityai/sd-vae-ft-ema
   denoiser:
     class_path: src.models.denoiser.improved_dit.DiT
     init_args:
diff --git a/configs/repa_improved_dit_xl.yaml b/configs/repa_improved_dit_xl.yaml
index 593c048..fc2a6e3 100644
--- a/configs/repa_improved_dit_xl.yaml
+++ b/configs/repa_improved_dit_xl.yaml
@@ -2,10 +2,10 @@
 seed_everything: true
 tags:
   exp: &exp repa_improved_dit_xlen22de6_512
-torch_hub_dir: /mnt/bn/wangshuai6/torch_hub
+torch_hub_dir: null
 huggingface_cache_dir: null
 trainer:
-  default_root_dir: /mnt/bn/wangshuai6/universal_flow_workdirs
+  default_root_dir: universal_flow_workdirs
   accelerator: auto
   strategy: auto
   devices: auto
@@ -40,7 +40,7 @@ model:
     class_path: src.models.vae.LatentVAE
     init_args:
       precompute: true
-      weight_path: /mnt/bn/wangshuai6/models/sd-vae-ft-ema/
+      weight_path: stabilityai/sd-vae-ft-ema
   denoiser:
     class_path: src.models.denoiser.improved_dit.DiT
     init_args: