Update README.md
Browse files
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.
|