ranjanrajib commited on
Commit
44d7ece
·
verified ·
1 Parent(s): 6365777

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -165,7 +165,7 @@ model = PeftModel.from_pretrained(
165
 
166
  ### Generating
167
 
168
-
169
  import re
170
  import torch
171
  from transformers import AutoModelForCausalLM, AutoTokenizer, set_seed
@@ -192,11 +192,12 @@ model = PeftModel.from_pretrained(
192
  response = tok.decode(out[0][inputs["input_ids"].shape[-1]:], skip_special_tokens=True)
193
  code = re.search(r"```python\n(.*?)```", response, re.S).group(1)
194
  print(code)
 
195
 
196
  ### What the model returned
197
 
198
  84 generated tokens, reproduced verbatim:
199
-
200
  import multiprocessing
201
  import math
202
 
@@ -209,6 +210,7 @@ model = PeftModel.from_pretrained(
209
  with multiprocessing.Pool() as pool:
210
  results = pool.map(factorial_task, numbers)
211
  return dict(results)
 
212
 
213
  ---
214
  ## Why Use This Model?
 
165
 
166
  ### Generating
167
 
168
+ ```python
169
  import re
170
  import torch
171
  from transformers import AutoModelForCausalLM, AutoTokenizer, set_seed
 
192
  response = tok.decode(out[0][inputs["input_ids"].shape[-1]:], skip_special_tokens=True)
193
  code = re.search(r"```python\n(.*?)```", response, re.S).group(1)
194
  print(code)
195
+ '''
196
 
197
  ### What the model returned
198
 
199
  84 generated tokens, reproduced verbatim:
200
+ ```python
201
  import multiprocessing
202
  import math
203
 
 
210
  with multiprocessing.Pool() as pool:
211
  results = pool.map(factorial_task, numbers)
212
  return dict(results)
213
+ '''
214
 
215
  ---
216
  ## Why Use This Model?