MDDDDR commited on
Commit
bc458d2
·
verified ·
1 Parent(s): 3651c9b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -20,6 +20,7 @@ datasets:
20
  ### Basic usage
21
  ```python
22
  from transformers import AutoTokenizer, AutoModelForCausalLM
 
23
 
24
  model_id = 'Llama-3.2-1B-Instruct-FFT-coder-python'
25
  tokenizer = AutoTokenizer.from_pretrained(model_id)
 
20
  ### Basic usage
21
  ```python
22
  from transformers import AutoTokenizer, AutoModelForCausalLM
23
+ import torch
24
 
25
  model_id = 'Llama-3.2-1B-Instruct-FFT-coder-python'
26
  tokenizer = AutoTokenizer.from_pretrained(model_id)