From e000b43cd4af21202c8b6917de6c45555f821358 Mon Sep 17 00:00:00 2001
From: zR <2448370773@qq.com>
Date: Tue, 27 Aug 2024 13:14:38 +0800
Subject: [PATCH] update ignore

---
 .gitignore                    | 3 ++-
 sat/configs/cogvideox_5b.yaml | 4 ++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/.gitignore b/.gitignore
index 695a4e2..9669d70 100644
--- a/.gitignore
+++ b/.gitignore
@@ -6,4 +6,5 @@ master_ip
 logs/
 *.DS_Store
 .idea
-output*
\ No newline at end of file
+output*
+test*
\ No newline at end of file
diff --git a/sat/configs/cogvideox_5b.yaml b/sat/configs/cogvideox_5b.yaml
index 160c5ad..6805dec 100644
--- a/sat/configs/cogvideox_5b.yaml
+++ b/sat/configs/cogvideox_5b.yaml
@@ -74,14 +74,14 @@ model:
           ucg_rate: 0.1
           target: sgm.modules.encoders.modules.FrozenT5Embedder
           params:
-            model_dir: "/share/official_pretrains/hf_home/t5-v1_1-xxl"
+            model_dir: "t5-v1_1-xxl"
             max_length: 226
 
   first_stage_config:
     target: vae_modules.autoencoder.VideoAutoencoderInferenceWrapper
     params:
       cp_size: 1
-      ckpt_path: "/share/official_pretrains/sat_home/cogvideox-5b-sat/vae/3d-vae.pt"
+      ckpt_path: "cogvideox-5b-sat/vae/3d-vae.pt"
       ignore_keys: [ 'loss' ]
 
       loss_config: