Merge pull request #62 from QwenLM/cyente-patch-17

Update CodeQwen1.5-base-repolevel-fim.py

examples/CodeQwen1.5-base-repolevel-fim.py

@@ -7,7 +7,7 @@
 
 # tokenize the input into tokens
 # set fim format into the corresponding file you need to infilling
-input_text = """<reponame>library-system
+input_text = """<repo_name>library-system
 <file_sep>library.py
 class Book:
     def __init__(self, title, author, isbn, copies):
@@ -97,7 +97,7 @@ def main():
 # The generated_ids include prompt_ids, so we only need to decode the tokens after prompt_ids.
 output_text = tokenizer.decode(generated_ids[len(model_inputs.input_ids[0]):], skip_special_tokens=True)
 
-print(f"Prompt: \n{input_text}\n\nGenerated text: \n{output_text}")
+print(f"Prompt: \n{input_text}\n\nGenerated text: \n{output_text.split('<file_sep>')[0]}")
 
 # the expected output as following:
 """