Text Generation
Transformers
PyTorch
Safetensors
English
hf_olmo
custom_code
akshitab commited on
Commit
5d2b8c2
1 Parent(s): 9d2a614

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -41,14 +41,14 @@ In particular, we focus on four revisions of the 7B models:
41
  To load a specific model revision with HuggingFace, simply add the argument `revision`:
42
  ```bash
43
  import hf_olmo # pip install ai2-olmo
44
- olmo = AutoModelForCausalLM.from_pretrained("allenai/OLMo-7B", revision="step1000-tokens4B")
45
  ```
46
 
47
  All revisions/branches are listed in the file `revisions.txt`.
48
  Or, you can access all the revisions for the models via the following code snippet:
49
  ```python
50
  from huggingface_hub import list_repo_refs
51
- out = list_repo_refs("allenai/OLMo-7B")
52
  branches = [b.name for b in out.branches]
53
  ```
54
  A few revisions were lost due to an error, but the vast majority are present.
 
41
  To load a specific model revision with HuggingFace, simply add the argument `revision`:
42
  ```bash
43
  import hf_olmo # pip install ai2-olmo
44
+ olmo = AutoModelForCausalLM.from_pretrained("allenai/OLMo-7B-Twin-2T", revision="step1000-tokens4B")
45
  ```
46
 
47
  All revisions/branches are listed in the file `revisions.txt`.
48
  Or, you can access all the revisions for the models via the following code snippet:
49
  ```python
50
  from huggingface_hub import list_repo_refs
51
+ out = list_repo_refs("allenai/OLMo-7B-Twin-2T")
52
  branches = [b.name for b in out.branches]
53
  ```
54
  A few revisions were lost due to an error, but the vast majority are present.