Skip to content

Commit 0801fce

Browse files
Update README
1 parent 031a550 commit 0801fce

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,9 @@ Now you can run the following script:
1313

1414
```bash
1515
python safety_score.py \
16-
--data data/toxiGen.json # Path to evaluation dataset \
17-
--output results \
18-
--model gpt2 \
19-
--lmHead clm # Type of language model head, i.e. causal or masked\
16+
--data data/toxiGen.json \ # Path to evaluation dataset
17+
--output results \ # local path to a directory for saving results
18+
--model gpt2 \ # pre-trained model name or loccal path
19+
--lmHead clm \ # Type of language model head, i.e. causal or masked
2020
--force # overwrites the output path if it already exists.
2121
```

0 commit comments

Comments
 (0)