├── README.md ├── arguments.py ├── data └── toxicity │ ├── .DS_Store │ ├── process.py │ ├── raw │ ├── .DS_Store │ ├── nontoxic_prompts-10k.jsonl │ ├── prompts.jsonl │ └── toy_prompt.jsonl │ ├── test.jsonl │ ├── train.jsonl │ └── val.jsonl ├── data_pool.py ├── environment.yml ├── main.py ├── model └── gpt2.py ├── perplexity.py ├── policy.py ├── reward.py ├── sample.py └── utils ├── constants.py ├── perspective_api.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/README.md -------------------------------------------------------------------------------- /arguments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/arguments.py -------------------------------------------------------------------------------- /data/toxicity/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/data/toxicity/.DS_Store -------------------------------------------------------------------------------- /data/toxicity/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/data/toxicity/process.py -------------------------------------------------------------------------------- /data/toxicity/raw/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/data/toxicity/raw/.DS_Store -------------------------------------------------------------------------------- /data/toxicity/raw/nontoxic_prompts-10k.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/data/toxicity/raw/nontoxic_prompts-10k.jsonl -------------------------------------------------------------------------------- /data/toxicity/raw/prompts.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/data/toxicity/raw/prompts.jsonl -------------------------------------------------------------------------------- /data/toxicity/raw/toy_prompt.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/data/toxicity/raw/toy_prompt.jsonl -------------------------------------------------------------------------------- /data/toxicity/test.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/data/toxicity/test.jsonl -------------------------------------------------------------------------------- /data/toxicity/train.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/data/toxicity/train.jsonl -------------------------------------------------------------------------------- /data/toxicity/val.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/data/toxicity/val.jsonl -------------------------------------------------------------------------------- /data_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/data_pool.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/environment.yml -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/main.py -------------------------------------------------------------------------------- /model/gpt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/model/gpt2.py -------------------------------------------------------------------------------- /perplexity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/perplexity.py -------------------------------------------------------------------------------- /policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/policy.py -------------------------------------------------------------------------------- /reward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/reward.py -------------------------------------------------------------------------------- /sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/sample.py -------------------------------------------------------------------------------- /utils/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/utils/constants.py -------------------------------------------------------------------------------- /utils/perspective_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/utils/perspective_api.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GXimingLu/Quark/HEAD/utils/utils.py --------------------------------------------------------------------------------