Is LLaMA 2 only used for Stage 1?

[djghosh13]

Is there any way to run the full pipeline replacing ChatGPT with LLaMA 2? It looks like if use_llama is set, LLaMA is used for only stage 1, but ChatGPT is still used for stage 2.

[DingchenYang99]

same question

[artemisp]

I created a pull request #6 addressing this issue. I hope it helps! Any feedback is also welcome to help me test it :)