Fix preference caption prefix/suffix

This commit is contained in:
rockerBOO
2024-07-05 13:06:13 -04:00
parent 1f355f77ac
commit aaac0f2ebd

View File

@@ -424,8 +424,8 @@ class BaseSubset:
self.token_warmup_step = token_warmup_step # NN<1ならN*max_train_stepsステップ目でタグの数が最大になる
self.preference = preference
self.preference_caption_prefix = non_preference_caption_prefix
self.preference_caption_suffix = non_preference_caption_suffix
self.preference_caption_prefix = preference_caption_prefix
self.preference_caption_suffix = preference_caption_suffix
self.non_preference_caption_prefix = non_preference_caption_prefix
self.non_preference_caption_suffix = non_preference_caption_suffix