RAIN
RAIN copied to clipboard
[ICLR'24] RAIN: Your Language Models Can Align Themselves without Finetuning
how to set max_new_tokens? I just could find 'maxlen' var in 'main.py', and it would cost too muck time for testing.
Thanks for your code! I am reaching out to discuss some observations I've made while utilizing your codebase. When executing the code snippet provided below, an error occurs and the...
I see the code and find that in the HH-RLHF dataset you use the red-team data for test. I want to know how the test scores are calculated? I didnt...
Thanks for your great work. I want to apply RAIN to Mistral ([https://huggingface.co/HuggingFaceH4/mistral-7b-sft-beta]) on HH test set. I only changed scripts of loading dataset in `main.py` (I want inference on...
when I run HH/main.py with the following setting: parser.add_argument('--modelname', type=str, default="./models/vicuna-7b-v1.5") I get an error: for epoch_test, batch_test in tqdm(enumerate(testloader)): Exception has occurred: RuntimeError stack expects each tensor to be...