separate source layer for steering vector.

This commit is contained in:
Henri Vasserman
2023-05-16 18:43:40 +03:00
parent 8388aaa604
commit c90059fba6
3 changed files with 11 additions and 3 deletions

View File

@@ -77,6 +77,7 @@ struct gpt_params {
std::string steering_sub;
float steering_mul = 1.0f;
int steering_layer = 15;
int steering_source = 2;
};
bool gpt_params_parse(int argc, char ** argv, gpt_params & params);