Update README.md

README.md

@@ -23,7 +23,8 @@ device = 'cuda'
 model = GPTNeoXForCausalLM.from_pretrained(model_name_or_path).to(device)
 tokenizer = AutoTokenizer.from_pretrained(model_name_or_path)
 
-instruction = '[Summarize the knowledge points in the code below]\n'		# instruction template
+# instruction template
+instruction = '[Summarize the knowledge points in the code below]\n'
 # any c-lang pieces you like, could be partial functions or statements
 input_content = '''```c
 int partition(int arr[], int low, int high) {
@@ -78,7 +79,7 @@ for i in range(30):		# maybe 20 times or less enough too
         pad_token_id=tokenizer.eos_token_id,
         max_new_tokens=32,
         do_sample=True,
-        temperature=2.0,                       # high temperature for diversity
+        temperature=2.0,  # high temperature for diversity
         top_p=0.95,
         top_k=30,
     )
@@ -87,5 +88,11 @@ for i in range(30):		# maybe 20 times or less enough too
 
 print(ans_dict)
 ### output as below, could take high-freq answers
-### {'Backtracking': 1, 'Heap': 1, 'Quick sort': 25, 'Recurrence': 2, 'Queue': 1}
+### {
+###     'Backtracking': 1,
+###     'Heap': 1,
+###     'Quick sort': 25,
+###     'Recurrence': 2,
+###     'Queue': 1
+###}
 ```