Update README.md
Browse files
README.md
CHANGED
|
@@ -90,9 +90,6 @@ If you want to use Hugging Face `transformers` to generate text, you can do some
|
|
| 90 |
|
| 91 |
```py
|
| 92 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 93 |
-
from accelerate import Accelerator #Use only GPU A100/L4
|
| 94 |
-
|
| 95 |
-
accelerator = Accelerator() #Use only GPU A100/L4
|
| 96 |
|
| 97 |
model_id = "EpistemeAI/Fireball-12B"
|
| 98 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
|
|
|
| 90 |
|
| 91 |
```py
|
| 92 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
|
|
|
|
|
|
|
|
| 93 |
|
| 94 |
model_id = "EpistemeAI/Fireball-12B"
|
| 95 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|