Update README.md
Browse files
README.md
CHANGED
@@ -32,13 +32,11 @@ We construct a comprehensive instruction dataset for detoxification of LLMs, and
|
|
32 |
|
33 |
This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
|
34 |
|
35 |
-
- **Developed by:**
|
36 |
-
- **Funded by [optional]:** [More Information Needed]
|
37 |
-
- **Shared by [optional]:** [More Information Needed]
|
38 |
- **Model type:** [More Information Needed]
|
39 |
-
- **Language(s) (NLP):**
|
40 |
-
- **License:**
|
41 |
-
- **Finetuned from model [optional]:**
|
42 |
|
43 |
### Model Sources [optional]
|
44 |
|
@@ -52,40 +50,24 @@ This is the model card of a 🤗 transformers model that has been pushed on the
|
|
52 |
|
53 |
<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
|
54 |
|
55 |
-
### Direct Use
|
56 |
-
|
57 |
-
<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
|
58 |
-
|
59 |
-
```python
|
60 |
-
from transformers import AutoTokenizer, AutoModelForCausalLM
|
61 |
-
|
62 |
-
tokenizer = AutoTokenizer.from_pretrained("SungJoo/llama2-7b-sft-detox")
|
63 |
-
model = AutoModelForCausalLM.from_pretrained("SungJoo/llama2-7b-sft-detox")
|
64 |
-
|
65 |
-
|
66 |
-
### Out-of-Scope Use
|
67 |
-
|
68 |
-
<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
|
69 |
-
|
70 |
-
[More Information Needed]
|
71 |
-
|
72 |
## Bias, Risks, and Limitations
|
73 |
|
74 |
<!-- This section is meant to convey both technical and sociotechnical limitations. -->
|
75 |
|
76 |
[More Information Needed]
|
77 |
|
78 |
-
### Recommendations
|
79 |
-
|
80 |
-
<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
|
81 |
-
|
82 |
-
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
|
83 |
|
84 |
## How to Get Started with the Model
|
85 |
|
86 |
Use the code below to get started with the model.
|
87 |
|
88 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
89 |
|
90 |
## Training Details
|
91 |
|
|
|
32 |
|
33 |
This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
|
34 |
|
35 |
+
- **Developed by:** Sungjoo Byun (Grace Byun)
|
|
|
|
|
36 |
- **Model type:** [More Information Needed]
|
37 |
+
- **Language(s) (NLP):** English
|
38 |
+
- **License:** Apache license 2.0
|
39 |
+
- **Finetuned from model [optional]:** meta-llama/Llama-2-7b-hf
|
40 |
|
41 |
### Model Sources [optional]
|
42 |
|
|
|
50 |
|
51 |
<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
|
52 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
53 |
## Bias, Risks, and Limitations
|
54 |
|
55 |
<!-- This section is meant to convey both technical and sociotechnical limitations. -->
|
56 |
|
57 |
[More Information Needed]
|
58 |
|
|
|
|
|
|
|
|
|
|
|
59 |
|
60 |
## How to Get Started with the Model
|
61 |
|
62 |
Use the code below to get started with the model.
|
63 |
|
64 |
+
|
65 |
+
```python
|
66 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM
|
67 |
+
|
68 |
+
tokenizer = AutoTokenizer.from_pretrained("SungJoo/llama2-7b-sft-detox")
|
69 |
+
model = AutoModelForCausalLM.from_pretrained("SungJoo/llama2-7b-sft-detox")
|
70 |
+
```
|
71 |
|
72 |
## Training Details
|
73 |
|