From b4a42bf444103dc3d321ff9fd221bd8862484423 Mon Sep 17 00:00:00 2001 From: alex Date: Tue, 18 Feb 2025 09:24:24 -0500 Subject: [PATCH] Update cosmos.sh --- cosmos.sh | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/cosmos.sh b/cosmos.sh index 24e3c11..4bbf34f 100644 --- a/cosmos.sh +++ b/cosmos.sh @@ -28,14 +28,12 @@ CHECKPOINT_MODELS=( ) CLIP_MODELS=( - "https://huggingface.co/comfyanonymous/cosmos_1.0_text_encoder_and_VAE_ComfyUI/resolve/main/text_encoders/oldt5_xxl_fp8_e4m3fn_scaled.safetensors" - "https://huggingface.co/comfyanonymous/cosmos_1.0_text_encoder_and_VAE_ComfyUI/resolve/main/text_encoders/oldt5_xxl_fp16.safetensors" ) DIFFUSION_MODELS=( #"https://huggingface.co/mcmonkey/cosmos-1.0/resolve/main/Cosmos-1_0-Diffusion-14B-Text2World.safetensors" #"https://huggingface.co/mcmonkey/cosmos-1.0/resolve/main/Cosmos-1_0-Diffusion-14B-Video2World.safetensors" - #"https://huggingface.co/mcmonkey/cosmos-1.0/resolve/main/Cosmos-1_0-Diffusion-7B-Text2World.safetensors" + "https://huggingface.co/mcmonkey/cosmos-1.0/resolve/main/Cosmos-1_0-Diffusion-7B-Text2World.safetensors" "https://huggingface.co/mcmonkey/cosmos-1.0/resolve/main/Cosmos-1_0-Diffusion-7B-Video2World.safetensors" ) @@ -59,6 +57,11 @@ CONTROLNET_MODELS=( ) +TEXT_ENCODERS=( + "https://huggingface.co/quarterturn/comfyui-cosmos-files/resolve/main/oldt5_xxl_fp16.safetensors" + "https://huggingface.co/quarterturn/comfyui-cosmos-files/resolve/main/oldt5_xxl_fp8_e4m3fn_scaled.safetensors" +) + ### DO NOT EDIT BELOW HERE UNLESS YOU KNOW WHAT YOU ARE DOING ### function provisioning_start() { @@ -90,6 +93,12 @@ function provisioning_start() { provisioning_get_models \ "${WORKSPACE}/storage/stable_diffusion/models/esrgan" \ "${ESRGAN_MODELS[@]}" + provisioning_get_models \ + "${WORKSPACE}/storage/stable_diffusion/models/text_encoders" \ + "${TEXT_ENCODERS[@]}" + provisioning_get_models \ + "${WORKSPACE}/storage/stable_diffusion/models/diffusion_models" \ + "${DIFFUSION_MODELS[@]}" provisioning_print_end }