Update README.md

README.md

@@ -53,7 +53,7 @@ A typical usage flow:
 import transformers
 import torch
 
-model_id = "ByteDance-Seed/Seed-Coder-8B-Base"
+model_id = "/mnt/bn/daoguang/ckpts/Bytedance/Doubao-Coder-base/P6Dense"
 
 pipeline = transformers.pipeline(
     "text-generation",
@@ -67,9 +67,9 @@ prefix = "def add_numbers(a, b):\n    "
 suffix = "\n    return result"
 
 # Combine prefix and suffix following the FIM format
-fim_input = "<|fim-suffix|>" + suffix + "<|fim-prefix|>" + prefix + "<|fim-middle|>"
+fim_input = '<[PLHD125_never_used_51bce0c785ca2f68081bfa7d91973934]>' + suffix + '<[PLHD124_never_used_51bce0c785ca2f68081bfa7d91973934]>' + prefix + '<[PLHD126_never_used_51bce0c785ca2f68081bfa7d91973934]>'
 
-output = pipeline(fim_input, max_new_tokens=100)
+output = pipeline(fim_input, max_new_tokens=512)
 print(output[0]["generated_text"])
 ```