eliebak HF staff commited on
Commit
ee389a1
1 Parent(s): 29ba436

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -35,7 +35,7 @@ pip install transformers
35
  #### Running the model on CPU/GPU/multi GPU
36
  * _Using full precision_
37
  ```python
38
- # pip install git+https://github.com/huggingface/transformers.git # TODO: merge PR to main
39
  from transformers import AutoModelForCausalLM, AutoTokenizer
40
  checkpoint = "HuggingFaceTB/SmolLM-360M"
41
  device = "cuda" # for GPU usage or "cpu" for CPU usage
 
35
  #### Running the model on CPU/GPU/multi GPU
36
  * _Using full precision_
37
  ```python
38
+ # pip install transformers
39
  from transformers import AutoModelForCausalLM, AutoTokenizer
40
  checkpoint = "HuggingFaceTB/SmolLM-360M"
41
  device = "cuda" # for GPU usage or "cpu" for CPU usage