osunlp/HIVE-MagicBrush

InstructPix2Pix checkpoint fine-tuned on MagicBrush.

1. Env Setup

To use the HIVE checkpoint fine-tuned on MagicBrush, set up env with following the original HIVE:

Download this checkpoint into checkpoints folder via

cd checkpoints
# recent best checkpoint:
wget https://huggingface.co/osunlp/HIVE-MagicBrush/resolve/main/MagicBrush-epoch-000130.ckpt

Citation

If you find this checkpoint useful, please consider citing our paper:

@inproceedings{Zhang2023MagicBrush,
        title={MagicBrush: A Manually Annotated Dataset for Instruction-Guided Image Editing},
        author={Kai Zhang and Lingbo Mo and Wenhu Chen and Huan Sun and Yu Su},
        booktitle={Advances in Neural Information Processing Systems},
        year={2023}
}