Skip to content

Commit

Permalink
chore: add RealVisXL_V4.0 models to download script
Browse files Browse the repository at this point in the history
  • Loading branch information
rickstaa committed May 3, 2024
1 parent 5fd6e76 commit b8dc520
Showing 1 changed file with 8 additions and 6 deletions.
14 changes: 8 additions & 6 deletions runner/dl_checkpoints.sh
Original file line number Diff line number Diff line change
Expand Up @@ -57,15 +57,15 @@ fi
printf "Downloading %s models...\n" "$MODE"
if [ "$MODE" = "alpha" ]; then
printf "\nDownloading unrestricted models...\n"

# Download text-to-image and image-to-image models.
huggingface-cli download ByteDance/SDXL-Lightning --include "*unet.safetensors" --exclude "*lora.safetensors*" --cache-dir models

# Download image-to-video models (token-gated).
printf "\nDownloading token-gated models...\n"
check_hf_auth
huggingface-cli download stabilityai/stable-video-diffusion-img2vid-xt-1-1 --include "*.fp16.safetensors" "*.json" --cache-dir models "${TOKEN_FLAG}"

printf "\nAlpha models downloaded successfully!\n"
else
# Download text-to-image and image-to-image models.
Expand All @@ -76,14 +76,16 @@ else
huggingface-cli download stabilityai/stable-diffusion-xl-base-1.0 --include "*.fp16.safetensors" "*.json" "*.txt" --exclude ".onnx" ".onnx_data" --cache-dir models
huggingface-cli download prompthero/openjourney-v4 --include "*.safetensors" "*.json" "*.txt" --exclude ".onnx" ".onnx_data" --cache-dir models
huggingface-cli download ByteDance/SDXL-Lightning --include "*unet.safetensors" --exclude "*lora.safetensors*" --cache-dir models

huggingface-cli download SG161222/RealVisXL_V4.0_Lightning --include "*.fp16.safetensors" "*.json" "*.txt" --exclude ".onnx" ".onnx_data" --cache-dir models
huggingface-cli download SG161222/RealVisXL_V4.0 --include "*.fp16.safetensors" "*.json" "*.txt" --exclude ".onnx" ".onnx_data" --cache-dir models

# Download image-to-video models.
huggingface-cli download stabilityai/stable-video-diffusion-img2vid-xt --include "*.fp16.safetensors" "*.json" --cache-dir models

# Download image-to-video models (token-gated).
printf "\nDownloading token-gated models...\n"
check_hf_auth
huggingface-cli download stabilityai/stable-video-diffusion-img2vid-xt-1-1 --include "*.fp16.safetensors" "*.json" --cache-dir models "${TOKEN_FLAG}"

printf "\nAll models downloaded successfully!\n"
fi

0 comments on commit b8dc520

Please sign in to comment.