Update README.md
Browse files
README.md
CHANGED
|
@@ -63,7 +63,7 @@ The following hyperparameters were used during training:
|
|
| 63 |
|
| 64 |
## Use this model
|
| 65 |
|
| 66 |
-
|
| 67 |
from transformers import AutoModelForCausalLM
|
| 68 |
from transformers import BitsAndBytesConfig
|
| 69 |
import torch
|
|
@@ -81,4 +81,4 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 81 |
torch_dtype=torch.bfloat16,
|
| 82 |
)
|
| 83 |
|
| 84 |
-
|
|
|
|
| 63 |
|
| 64 |
## Use this model
|
| 65 |
|
| 66 |
+
```
|
| 67 |
from transformers import AutoModelForCausalLM
|
| 68 |
from transformers import BitsAndBytesConfig
|
| 69 |
import torch
|
|
|
|
| 81 |
torch_dtype=torch.bfloat16,
|
| 82 |
)
|
| 83 |
|
| 84 |
+
```
|