#!/bin/bash

# 319MB ???
mkdir -p ./input/assets/vae/
wget -nc "https://huggingface.co/RunDiffusion/Juggernaut-X-v10/resolve/main/vae/diffusion_pytorch_model.bin" -O ./input/assets/vae/diffusion_pytorch_model.bin

# Large (9,6GB) Modell?!?
mkdir -p ./input/assets/unet/
wget -nc "https://huggingface.co/RunDiffusion/Juggernaut-X-v10/resolve/main/unet/diffusion_pytorch_model.bin" -O ./input/assets/unet/diffusion_pytorch_model.bin

# 2,78GB
mkdir -p ./input/assets/text_encoder_2/
wget -nc "https://huggingface.co/RunDiffusion/Juggernaut-X-v10/resolve/main/text_encoder_2/pytorch_model.bin" -O ./input/assets/text_encoder_2/pytorch_model.bin

# 492MB
mkdir -p ./input/assets/text_encoder/
wget -nc "https://huggingface.co/RunDiffusion/Juggernaut-X-v10/resolve/main/text_encoder/pytorch_model.bin" -O ./input/assets/text_encoder/pytorch_model.bin

podman build -t tti-function .