LLM4APR
/

CodeLlama-70B_for_NMT

Model card Files Files and versions

LLM4APR commited on Mar 28, 2024

Commit

af04ce9

·

verified ·

1 Parent(s): 3fbb0a5

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -103,7 +103,7 @@ generated_ids = model.generate(
 for generated_id in generated_ids:
     generated_text = tokenizer.decode(generated_id, skip_special_tokens=False)
     patch = generated_text.split(E_INST)[1]
-    patch = text.replace(tokenizer.eos_token,'')
     print(patch)

 for generated_id in generated_ids:
     generated_text = tokenizer.decode(generated_id, skip_special_tokens=False)
     patch = generated_text.split(E_INST)[1]
+    patch = patch.replace(tokenizer.eos_token,'')
     print(patch)