Update README.md
Browse files
README.md
CHANGED
@@ -79,11 +79,14 @@ Training Results
|
|
79 |
-Supports Alpaca , ChatML and maybe others too
|
80 |
|
81 |
|
82 |
-
```
|
|
|
|
|
83 |
|
84 |
|
85 |
|
86 |
-
```
|
|
|
87 |
import torch
|
88 |
|
89 |
model_name = "fhai50032/RolePlayLake-7B-Toxic"
|
|
|
79 |
-Supports Alpaca , ChatML and maybe others too
|
80 |
|
81 |
|
82 |
+
```python
|
83 |
+
pip install -qU transformers accelerate bitsandbytes
|
84 |
+
```
|
85 |
|
86 |
|
87 |
|
88 |
+
```python
|
89 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, pipeline
|
90 |
import torch
|
91 |
|
92 |
model_name = "fhai50032/RolePlayLake-7B-Toxic"
|