TextDiffuser / text-to-image.sh
watermelon23's picture
update
e8dca02
raw
history blame
221 Bytes
CUDA_VISIBLE_DEVICES=0 python inference.py \
--mode="text-to-image" \
--resume_from_checkpoint="textdiffuser-ckpt/diffusion_backbone" \
--prompt="A sign that says 'Hello'" \
--output_dir="./output" \
--vis_num=4