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
No known key found for this signature in database
GPG key ID: 2995FC0F58B1A986
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);