ByteDance-Seed
/

Seed-Coder-8B-Base

Text Generation

text-generation-inference

Model card Files Files and versions

yuyuzhang commited on Apr 28

Commit

0d07384

·

verified ·

1 Parent(s): 2b4d112

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -67,7 +67,7 @@ prefix = "def add_numbers(a, b):\n    "
 suffix = "\n    return result"
 # Combine prefix and suffix following the FIM format
-fim_input = '<[PLHD125_never_used_51bce0c785ca2f68081bfa7d91973934]>' + suffix + '<[PLHD124_never_used_51bce0c785ca2f68081bfa7d91973934]>' + prefix + '<[PLHD126_never_used_51bce0c785ca2f68081bfa7d91973934]>'
 output = pipeline(fim_input, max_new_tokens=512)
 print(output[0]["generated_text"])

 suffix = "\n    return result"
 # Combine prefix and suffix following the FIM format
+fim_input = '<[fim-suffix]>' + suffix + '<[fim-prefix]>' + prefix + '<[fim-middle]>'
 output = pipeline(fim_input, max_new_tokens=512)
 print(output[0]["generated_text"])