source ~/.bashrc
# conda activate mace-update-v4-ft
conda activate mace-update-v5


# export OPENAI_API_KEY=<your-api-key>

PREFIX=/data/cluster_name/scratch/$(whoami)/projects/MACE-Update

LAION_DIR=${PREFIX}/data/laion400m
UNLABELED_DIR=/data/cluster_name/gobi1/data/laion400m
EMBEDDINGS_DIR=$UNLABELED_DIR/embeddings

mkdir -p $EMBEDDINGS_DIR

clip-retrieval inference --input_dataset "$LAION_DIR" --output_folder "$EMBEDDINGS_DIR"