From 6468a2d6ed2db748f4d2d7fb83144f45e852f5e7 Mon Sep 17 00:00:00 2001 From: Hyunho Kim <52307315+XEL-Maker@users.noreply.github.com> Date: Mon, 10 Feb 2025 09:03:51 +0900 Subject: [PATCH 1/2] Update README.md Fix W&B project name issue: Use WANDB_PROJECT to apply correctly --- cosmos1/models/autoregressive/nemo/post_training/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cosmos1/models/autoregressive/nemo/post_training/README.md b/cosmos1/models/autoregressive/nemo/post_training/README.md index ef0f7e18..8ecedb34 100644 --- a/cosmos1/models/autoregressive/nemo/post_training/README.md +++ b/cosmos1/models/autoregressive/nemo/post_training/README.md @@ -147,7 +147,7 @@ Complete the following steps to post-train the Cosmos-1.0-Autoregressive-4B mode # Optionally, you can monitor training progress with Weights and Biases (wandb). export WANDB_API_KEY="" - export WANDB_PROJECT_NAME="cosmos-autoregressive-nemo-finetuning" + export WANDB_PROJECT="cosmos-autoregressive-nemo-finetuning" export WANDB_RUN_ID="cosmos_autoregressive_4b_finetune" ``` 2. Run the following command for Cosmos-1.0-Autoregressive-4B post-training: From bc40871d7a3b69b4abc697d537cff3a48a353ec6 Mon Sep 17 00:00:00 2001 From: Hyunho Kim <52307315+XEL-Maker@users.noreply.github.com> Date: Mon, 10 Feb 2025 09:04:33 +0900 Subject: [PATCH 2/2] Update README.md Fix W&B project name issue: Use WANDB_PROJECT to apply correctly --- cosmos1/models/diffusion/nemo/post_training/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cosmos1/models/diffusion/nemo/post_training/README.md b/cosmos1/models/diffusion/nemo/post_training/README.md index ba8c3ded..0778546d 100644 --- a/cosmos1/models/diffusion/nemo/post_training/README.md +++ b/cosmos1/models/diffusion/nemo/post_training/README.md @@ -150,7 +150,7 @@ Complete the following steps to post-train the Cosmos-1.0-Diffusion-7B-Text2Worl # Optionally, you can monitor training progress with Weights and Biases (wandb). export WANDB_API_KEY="" - export WANDB_PROJECT_NAME="cosmos-diffusion-nemo-post-training" + export WANDB_PROJECT="cosmos-diffusion-nemo-post-training" export WANDB_RUN_ID="cosmos_diffusion_7b_text2world_finetune" ``` 2. Run the following command for Cosmos-Diffusion-Text2World-7B general post-training: