diff --git a/README.md b/README.md index d5e243a..d66bfc6 100644 --- a/README.md +++ b/README.md @@ -64,7 +64,7 @@ formatted_prompt = apply_chat_template( ) # Generate output -output = generate(model, processor, image, formatted_prompt, verbose=False) +output = generate(model, processor, formatted_prompt, image, verbose=False) print(output) ``` @@ -102,7 +102,7 @@ formatted_prompt = apply_chat_template( processor, config, prompt, num_images=len(images) ) -output = generate(model, processor, images, formatted_prompt, verbose=False) +output = generate(model, processor, formatted_prompt, images, verbose=False) print(output) ```