Typo in the README.md for janus-flow text-to-image tutorial

[CiaoHe]

...
if step == 0:
    outputs = vl_gpt.language_model.model(inputs_embeds=llm_emb, 
                                     use_cache=True, 
                                     attention_mask=attention_mask,
                                     past_key_values=None)
    past_key_values = []
    for kv_cache in past_key_values: # here, should be `outputs.past_key_values` ?
        k, v = kv_cache[0], kv_cache[1]
        past_key_values.append((k[:, :, :inputs_embeds.shape[1], :], v[:, :, :inputs_embeds.shape[1], :]))
    past_key_values = tuple(past_key_values)
...

[clivinn-shla81092]

Hello, we tried to solve the issue.

This is what we did:

Updated the generate function to correctly handle past_key_values from the model outputs. This fixes the issue where past_key_values was incorrectly initialized and not using the output from the model.

You can review changes in this commit: clivinn-shla81092@25e1a7d.

Caution

Disclaimer: The concept of solution was created by AI and you should never copy paste this code before you check the correctness of generated code. Solution might not be complete, you should use this code as an inspiration only.

---

Latta AI seeks to solve problems in open source projects as part of its mission to support developers around the world. Learn more about our mission at https://latta.ai/ourmission . If you no longer want Latta AI to attempt solving issues on your repository, you can block this account.