BAAI
/

Emu3-Chat

ryanzhangfan commited on 29 days ago

Commit

d9cb6ff

•

1 Parent(s): a7b7fab

Update README.md

Files changed (1) hide show

README.md CHANGED Viewed

@@ -89,7 +89,7 @@ GENERATION_CONFIG = GenerationConfig(
 outputs = model.generate(
     inputs.input_ids.to("cuda:0"),
     GENERATION_CONFIG,
-    attention_mask=pos_inputs.attention_mask.to("cuda:0"),
 )
 outputs = outputs[:, inputs.input_ids.shape[-1]:]

 outputs = model.generate(
     inputs.input_ids.to("cuda:0"),
     GENERATION_CONFIG,
+    attention_mask=inputs.attention_mask.to("cuda:0"),
 )
 outputs = outputs[:, inputs.input_ids.shape[-1]:]