#!/bin/bash
# Start docker container in WSL to debug locally
NAME=T2A
TEXT_ENCODERS=/mnt/e/weights/flux_text_encoders
UNET_SAFETENSORS=/mnt/e/weights/FLUX.1-dev/flux1-dev.safetensors
VEA_SAFETENSORS=/mnt/e/weights/FLUX.1-dev/ae.safetensors 
IMAGE_NAME=text2autochrome
docker run --gpus all -it --rm \
  --name=$NAME \
  -v $TEXT_ENCODERS:/workspace/autochrome-gen-cui/models/text_encoders \
  -v $UNET_SAFETENSORS:/workspace/autochrome-gen-cui/models/unet/flux1-dev.safetensors \
  -v $VEA_SAFETENSORS:/workspace/autochrome-gen-cui/models/vae/ae.safetensors  \
  $IMAGE_NAME