Hugging Face's logo

ibm-granite
/
granite-3.1-3b-a800m-base

Text Generation
Transformers
Safetensors
granitemoe
language
granite-3.1
Model card Files Community
rpand002 commited on
Commit
3e0c59c
•
1 Parent(s): 2a5a4b1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -41,7 +41,7 @@ Then, copy the code snippet below to run the example.
41
  ```python
42
  from transformers import AutoModelForCausalLM, AutoTokenizer
43
  device = "auto"
44
- model_path = "ibm-granite/Granite-3.1-3B-A800M-base"
45
  tokenizer = AutoTokenizer.from_pretrained(model_path)
46
  # drop device_map if running on CPU
47
  model = AutoModelForCausalLM.from_pretrained(model_path, device_map=device)
 
41
  ```python
42
  from transformers import AutoModelForCausalLM, AutoTokenizer
43
  device = "auto"
44
+ model_path = "ibm-granite/granite-3.1-3b-a800m-base"
45
  tokenizer = AutoTokenizer.from_pretrained(model_path)
46
  # drop device_map if running on CPU
47
  model = AutoModelForCausalLM.from_pretrained(model_path, device_map=device)