pnevskaiaan commited on
Commit
7a5e67c
·
verified ·
1 Parent(s): e26ddf8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -2,6 +2,7 @@
2
  license: apache-2.0
3
  ---
4
 
 
5
  import torch
6
  from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer
7
  import os
@@ -57,4 +58,4 @@ tokenizer = PreTrainedTokenizerFast(
57
 
58
  print(f"Saved tiny model to {save_dir}")
59
  print("Files:", os.listdir(save_dir))
60
-
 
2
  license: apache-2.0
3
  ---
4
 
5
+ ```python
6
  import torch
7
  from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer
8
  import os
 
58
 
59
  print(f"Saved tiny model to {save_dir}")
60
  print("Files:", os.listdir(save_dir))
61
+ ```