deepseekmirror/Janus
1
0
Fork 0
mirror of https://github.com/deepseek-ai/Janus.git synced 2025-04-19 10:09:00 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: image generate in demo/app_janusflow.py

Browse Source
This commit is contained in:
yangmeng 2025-01-28 23:43:18 +08:00
parent 17bc41245a
commit 877c778c0e
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
demo/app_janusflow.py
View File

@ -117,8 +117,7 @@ def generate(
if step == 0:
outputs = vl_gpt.language_model.model(inputs_embeds=llm_emb,
use_cache=True,
attention_mask=attention_mask,
past_key_values=None)
attention_mask=attention_mask)
past_key_values = []
for kv_cache in past_key_values:
k, v = kv_cache[0], kv_cache[1]
@ -127,8 +126,7 @@ def generate(
else:
outputs = vl_gpt.language_model.model(inputs_embeds=llm_emb,
use_cache=True,
attention_mask=attention_mask,
past_key_values=past_key_values)
attention_mask=attention_mask)
hidden_states = outputs.last_hidden_state
# transform hidden_states back to v