set -xeuo pipefail

git clone -q https://github.com/khangtruong-ui/ai-detection-unet-gpu
cd ai-detection-unet-gpu
pip install -q uv
uv pip install -e .[diffusion]
hf auth login --token $HF_TOKEN
# wget -q https://huggingface.co/KhangTruong/Temporary/resolve/main/outputs.zip
# unzip -qq outputs.zip
# rm outputs.zip

# sid-train --config configs/experiments/sam3-qlora/sam3_qlora_beyondthebrush_r16.yaml \
#   --override data.batch_size=32 \
#  project.name=sid_sam3_qlora_coco_inpainted_r16 project.output_dir=outputs/RUN/sam3_qlora_coco_inpainted_r16

sid-train --configs configs/experiments/diffusion_diff/default.yaml configs/experiments/diffusion_diff_v2/default.yaml --override data.batch_size=32 training.epochs=40 data.streaming=False

hf upload KhangTruong/Temporary outputs joint_big_checkpoint