mbkim commited on
Commit
9e155b9
·
verified ·
1 Parent(s): 92e6db3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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)