songafu

JanusFlow示例代码修复建议

3

在使用JanusFlow的过程中，发现示例代码中的文生图部分在当前最新的Transformer版本（>=4.48.0）下无法正常运行且示例代码可能存在代码缺陷，建议修复如下： （1）JanusFlow示例代码（文生图）中数据流的处理，存在变量引用逻辑错误 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 outputs.past_key_values: #should be outputs.past_key_values k, v = kv_cache[0], kv_cache[1] past_key_values.append((k[:, :, :inputs_embeds.shape[1],...