search

ExplainableML / Vision_by_Language

[ICLR 2024] Official repository for "Vision-by-Language for Training-Free Compositional Image Retrieval"
MIT License
37 stars 2 forks source link

use llama #3

Closed zjjJacinthe closed 4 months ago

zjjJacinthe commented 5 months ago

I want to replace the gpt-turbo-3.5 model in the code with a llama series model. What do I need to do?

sgk98 commented 5 months ago

Hey, The gpt-3.5-turbo model call is being made here: https://github.com/ExplainableML/Vision_by_Language/blob/master/src/utils.py#L180. You would need to replace the API call with your call to a LLaMa model, the prompt (with the in-context examples) is already available here.