Dataset is now directly tokenized to avoid further post-processing of the trainer; locCustomPromptTemplate to specify a custom Jinja template; showChatTemplate for debugging;

This commit is contained in:
extio-de
2024-12-26 21:42:59 +01:00
parent cb9e62dbac
commit 63192f2bc0
5 changed files with 26 additions and 6 deletions

View File

@@ -97,10 +97,10 @@ class Trainer:
if context.trSchedulerType:
sftArgs["lr_scheduler_type"] = context.trSchedulerType
sftConfig = SFTConfig(
dataset_text_field="text",
output_dir=context.locWorkdir,
save_strategy="no",
#neftune_noise_alpha=5,
dataset_kwargs={'skip_prepare_dataset': True},
**sftArgs
)