mrm8488 commited on
Commit
f21c0d5
β€’
1 Parent(s): 395ae23

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -79,9 +79,9 @@ WIP
79
  ### Example of usage πŸ‘©β€πŸ’»
80
  ```py
81
  import torch
82
- from transformers import AutoModelForCausalLM, AutoTokenizer, AutoTokenizer
83
 
84
- model_id = "mrm8488/falcoder-7b"
85
 
86
  tokenizer = AutoTokenizer.from_pretrained(model_id)
87
 
 
79
  ### Example of usage πŸ‘©β€πŸ’»
80
  ```py
81
  import torch
82
+ from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
83
 
84
+ model_id = "mrm8488/llama-2-coder-7b"
85
 
86
  tokenizer = AutoTokenizer.from_pretrained(model_id)
87