Update README.md
Browse files
README.md
CHANGED
@@ -17,7 +17,7 @@ This is a [BERTIN GPT-J-6B](https://huggingface.co/bertin-project/bertin-gpt-j-6
|
|
17 |
```python
|
18 |
from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig, pipeline
|
19 |
|
20 |
-
base_model = "bertin-gpt-j-6B-alpaca"
|
21 |
tokenizer = AutoTokenizer.from_pretrained(base_model)
|
22 |
model = AutoModelForCausalLM.from_pretrained(
|
23 |
base_model,
|
|
|
17 |
```python
|
18 |
from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig, pipeline
|
19 |
|
20 |
+
base_model = "bertin-project/bertin-gpt-j-6B-alpaca"
|
21 |
tokenizer = AutoTokenizer.from_pretrained(base_model)
|
22 |
model = AutoModelForCausalLM.from_pretrained(
|
23 |
base_model,
|