From 7168ad4b6b090ce3a23d70a57ac0726c3c0f8913 Mon Sep 17 00:00:00 2001 From: AlessioGalluccio Date: Wed, 8 Jan 2025 00:02:10 +0100 Subject: [PATCH] Simplify README instructions. Signed-off-by: AlessioGalluccio --- cosmos1/models/diffusion/README.md | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/cosmos1/models/diffusion/README.md b/cosmos1/models/diffusion/README.md index 3b91bef5..572e8de6 100644 --- a/cosmos1/models/diffusion/README.md +++ b/cosmos1/models/diffusion/README.md @@ -40,9 +40,12 @@ huggingface-cli login 4. Download the Cosmos model weights from [Hugging Face](https://huggingface.co/collections/nvidia/cosmos-6751e884dc10e013a0a0d8e6): ```bash -PYTHONPATH=$(pwd) python cosmos1/scripts/download_diffusion.py --model_sizes 7B 14B --model_types Text2World Video2World +PYTHONPATH=$(pwd) python cosmos1/scripts/download_diffusion.py --model_sizes 7B --model_types Text2World Video2World +``` +If you would like to use the 14B model as well, run the following command: +```bash +PYTHONPATH=$(pwd) python cosmos1/scripts/download_diffusion.py --model_sizes 14B --model_types Text2World Video2World ``` - 5. The downloaded files should be in the following structure: ```