OSU-NLP-Group / MagicBrush

[NeurIPS'23] "MagicBrush: A Manually Annotated Dataset for Instruction-Guided Image Editing".
https://osu-nlp-group.github.io/MagicBrush/
Other
308 stars 14 forks source link

Multi-turn inference code? #7

Closed yuzhou914 closed 1 year ago

yuzhou914 commented 1 year ago

Thank you very much for that great work, and I wonder if you can release the single-turn and multi-turn inference code?

drogozhang commented 1 year ago

Hi, thanks for your kind words!

We have just uploaded the inference script named edit_cli_single_multi_turn.py. Please note that it was designed to be used with the dataset format we shared in the OneDrive link. We plan to add support for the Huggingface dataset format in the future.

Let me know if you have any other questions.

yuzhou914 commented 1 year ago

Yes, thank you very much for your update.