Skip to content

ArmanZarei/Mitigating-T2I-Comp-Issues

Repository files navigation

Mitigating-T2I-Compositionality-Issues

Training WiCLP sample script

python train_text_emb_projection.py \
    --output_dir="final_experiments/window_based_linear_projection/sd2_1/color" \
    --stable_diffusion_checkpoint="stabilityai/stable-diffusion-2-1" \
    --image_size=768 \
    --evaluation_batch_size=10 \
    --train_batch_size=4 \
    --projector_type="window_aware_linear" \
    --projection_window_size=5 \
    --train_steps=25000 \
    --validation_steps=1000 \
    --wandb_run_name="Color dataset - SD v2.1"

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published