Hugging Face's logo

ibm-granite
/
granite-3.1-2b-base

Text Generation
Transformers
Safetensors
granite
language
granite-3.1
Inference Endpoints
Model card Files Community
rpand002 commited on
Commit
31f1cf9
·
verified ·
1 Parent(s): 87dbf20

Update README.md

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