File size: 1,308 Bytes
980c5dd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
#!/bin/bash
# Configuration
install_dir=~/ # Define the installation directory
do_aws_setup=true # Flag to control AWS setup
# Create necessary directories
mkdir -p "$install_dir/models"
mkdir -p "$install_dir/datasets"
# Clone and set up naifu-diffusion
cd "$install_dir" || exit
git clone https://github.com/Mikubill/naifu-diffusion nd && cd nd
git checkout sgm
pip install -r requirements.txt xformers prodigyopt
# Download parameter calculator
param_calc="https://huggingface.co/trojblue/nd-param-calculator/resolve/main/nd_param_calculator_latest.ipynb"
wget "$param_calc"
# AWS setup
if [ "$do_aws_setup" = true ]; then
echo "Setting up AWS..."
# Check if aws command exists
if ! command -v aws &> /dev/null; then
echo "Installing AWS CLI..."
curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o "awscliv2.zip"
sudo apt install unzip && unzip awscliv2.zip
sudo ./aws/install
rm awscliv2.zip
else
echo "AWS CLI is already installed."
fi
# AWS S3 copy command
cd "$install_dir/models" || exit
aws s3 cp s3://bucket-external/trainer_opt/sdxl-fulldan-run-continue/checkpoint/nd-epoch-09.ckpt fd5me9.ckpt
else
echo "Skipping AWS setup..."
fi
# Add any additional setup or installation commands below
|