bhatta1 commited on
Commit
9f4d850
·
verified ·
1 Parent(s): dc0380a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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.seed1` model.
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.seed1"
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)