Merge pull request #6 from li-plus/fix-demo

Fix example code in README
This commit is contained in:
XCL 2024-12-16 15:21:46 +08:00 committed by GitHub
commit 2a99af7eea
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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,