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: