Fix example code in README

This commit is contained in:
Jiahao Li 2024-12-16 14:46:55 +08:00
parent b3a4b9bbdd
commit 85cdab0f5f

View File

@ -155,7 +155,7 @@ prepare_inputs = vl_chat_processor(
inputs_embeds = vl_gpt.prepare_inputs_embeds(**prepare_inputs)
# run the model to get the response
outputs = vl_gpt.language_model.generate(
outputs = vl_gpt.language.generate(
inputs_embeds=inputs_embeds,
attention_mask=prepare_inputs.attention_mask,
pad_token_id=tokenizer.eos_token_id,