Skip to content

Commit

Permalink
chore: fix alpha models (#104)
Browse files Browse the repository at this point in the history
This commit ensures that the download script downloads the right alpha
models when the `alpha` flag is specified.
  • Loading branch information
rickstaa authored Jun 13, 2024
1 parent 45687a7 commit 8c8c69d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion runner/dl_checkpoints.sh
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ function download_alpha_models() {
printf "\nDownloading unrestricted models...\n"

# Download text-to-image and image-to-image models.
huggingface-cli download SG161222/RealVisXL_V4.0_Lightning --include "*.fp16.safetensors" "*.json" "*.txt" --exclude ".onnx" ".onnx_data" --cache-dir models
huggingface-cli download ByteDance/SDXL-Lightning --include "*unet.safetensors" --cache-dir models
huggingface-cli download timbrooks/instruct-pix2pix --include "*.fp16.safetensors" "*.json" "*.txt" --cache-dir models

Expand All @@ -50,7 +51,6 @@ function download_all_models() {
huggingface-cli download runwayml/stable-diffusion-v1-5 --include "*.fp16.safetensors" "*.json" "*.txt" --exclude ".onnx" ".onnx_data" --cache-dir models
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 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.
Expand Down

0 comments on commit 8c8c69d

Please sign in to comment.