Update README.md
Browse files
README.md
CHANGED
|
@@ -22,7 +22,7 @@ Input: "[INST] Evaluate the toxicity of answer for given questions. You should o
|
|
| 22 |
Output: GT_Label (Safe or Unsafe)
|
| 23 |
```
|
| 24 |
Please initialize this model as
|
| 25 |
-
|
| 26 |
peft_config = LoraConfig(
|
| 27 |
lora_alpha=16,
|
| 28 |
lora_dropout=0.1,
|
|
@@ -34,8 +34,8 @@ peft_config = LoraConfig(
|
|
| 34 |
|
| 35 |
tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
|
| 36 |
model = AutoPeftModelForCausalLM.from_pretrained(model_path, config=peft_config, device_map="auto")
|
|
|
|
| 37 |
|
| 38 |
-
'''
|
| 39 |
### LifeTox Sources
|
| 40 |
|
| 41 |
- **Paper:** [arxiv](https://arxiv.org/abs/2311.09585)
|
|
|
|
| 22 |
Output: GT_Label (Safe or Unsafe)
|
| 23 |
```
|
| 24 |
Please initialize this model as
|
| 25 |
+
```
|
| 26 |
peft_config = LoraConfig(
|
| 27 |
lora_alpha=16,
|
| 28 |
lora_dropout=0.1,
|
|
|
|
| 34 |
|
| 35 |
tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
|
| 36 |
model = AutoPeftModelForCausalLM.from_pretrained(model_path, config=peft_config, device_map="auto")
|
| 37 |
+
```
|
| 38 |
|
|
|
|
| 39 |
### LifeTox Sources
|
| 40 |
|
| 41 |
- **Paper:** [arxiv](https://arxiv.org/abs/2311.09585)
|