README.md multimodal params fix (#967)
multi modal params fix: add logits = True -> to make llava work
This commit is contained in:
parent
c1d92ce680
commit
6bbeea07ae
1 changed files with 2 additions and 1 deletions
|
@ -266,7 +266,8 @@ Then you'll need to use a custom chat handler to load the clip model and process
|
||||||
>>> llm = Llama(
|
>>> llm = Llama(
|
||||||
model_path="./path/to/llava/llama-model.gguf",
|
model_path="./path/to/llava/llama-model.gguf",
|
||||||
chat_handler=chat_handler,
|
chat_handler=chat_handler,
|
||||||
n_ctx=2048 # n_ctx should be increased to accomodate the image embedding
|
n_ctx=2048, # n_ctx should be increased to accomodate the image embedding
|
||||||
|
logits_all=True,# needed to make llava work
|
||||||
)
|
)
|
||||||
>>> llm.create_chat_completion(
|
>>> llm.create_chat_completion(
|
||||||
messages = [
|
messages = [
|
||||||
|
|
Loading…
Reference in a new issue