Update cosmos.sh

This commit is contained in:
2025-02-18 09:24:24 -05:00
parent f67f95aca5
commit b4a42bf444

View File

@@ -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
}