Update README.md
Browse files
README.md
CHANGED
|
@@ -41,7 +41,7 @@ pip install transformers
|
|
| 41 |
```python
|
| 42 |
# pip install transformers
|
| 43 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 44 |
-
checkpoint = "GoudaAI/
|
| 45 |
device = "cuda" # for GPU usage or "cpu" for CPU usage
|
| 46 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
|
| 47 |
# for multiple GPUs install accelerate and do `model = AutoModelForCausalLM.from_pretrained(checkpoint, device_map="auto")`
|
|
|
|
| 41 |
```python
|
| 42 |
# pip install transformers
|
| 43 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 44 |
+
checkpoint = "GoudaAI/Cheddar-2.0-Instruct"
|
| 45 |
device = "cuda" # for GPU usage or "cpu" for CPU usage
|
| 46 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
|
| 47 |
# for multiple GPUs install accelerate and do `model = AutoModelForCausalLM.from_pretrained(checkpoint, device_map="auto")`
|