Update README.md
Browse files
README.md
CHANGED
@@ -37,7 +37,7 @@ This model is trained on 350B tokens of English FineWeb V1.1.0 data and is not i
|
|
37 |
|
38 |
**Generation**:
|
39 |
|
40 |
-
This is a simple example of how to use `GneissWeb.7B_ablation_model_on_350B_GneissWeb.
|
41 |
|
42 |
Install the following libraries:
|
43 |
|
@@ -51,7 +51,7 @@ Then, copy the code snippet below to run the example.
|
|
51 |
```python
|
52 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
53 |
device = "auto"
|
54 |
-
model_path = "ibm-granite/GneissWeb.7B_ablation_model_on_350B_GneissWeb.
|
55 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
56 |
# drop device_map if running on CPU
|
57 |
model = AutoModelForCausalLM.from_pretrained(model_path, device_map=device)
|
|
|
37 |
|
38 |
**Generation**:
|
39 |
|
40 |
+
This is a simple example of how to use `GneissWeb.7B_ablation_model_on_350B_GneissWeb.seed2` model.
|
41 |
|
42 |
Install the following libraries:
|
43 |
|
|
|
51 |
```python
|
52 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
53 |
device = "auto"
|
54 |
+
model_path = "ibm-granite/GneissWeb.7B_ablation_model_on_350B_GneissWeb.seed2"
|
55 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
56 |
# drop device_map if running on CPU
|
57 |
model = AutoModelForCausalLM.from_pretrained(model_path, device_map=device)
|