About the arguments specified for inference

CyberAgentAILab / OpenCOLE

OpenCOLE: Towards Reproducible Automatic Graphic Design Generation [Inoue+, CVPRW2024 (GDUG)]

Apache License 2.0

47 stars 5 forks source link

About the arguments specified for inference #7

Closed pacifinapacific closed 4 months ago

pacifinapacific commented 4 months ago

Thank you for publishing the code. I have a question because I do not fully understand the contents of the repository. What are and required for inference? Are they included in the repository or in the contents of the hugging face? It would be helpful if you could share the demo code for inference

naoto0804 commented 4 months ago

Hi, thank you for your interest in our repo.

INPUT_HFDS is typically a cyberagent/opencole, which is hosted in the HuggingFace dataset hub.
DETAIL_DIR indicates the directory containing the predicted results of opencole.inference_launcher.k_shot_llm (see doc)

Hope it helps!

pacifinapacific commented 4 months ago

I got it. Thank you very much!