diff --git a/src/autotrain/trainers/dreambooth/__main__.py b/src/autotrain/trainers/dreambooth/__main__.py index 96eb3a8655..55c6b658cc 100644 --- a/src/autotrain/trainers/dreambooth/__main__.py +++ b/src/autotrain/trainers/dreambooth/__main__.py @@ -181,7 +181,7 @@ def train(config): def save_model_hook(models, weights, output_dir): # there are only two options here. Either are just the unet attn processor layers - # or there are the unet and text encoder atten layers + # or there are the unet and text encoder attn layers unet_lora_layers_to_save = None text_encoder_lora_layers_to_save = [] diff --git a/src/autotrain/trainers/dreambooth/datasets.py b/src/autotrain/trainers/dreambooth/datasets.py index 791c46d510..8121094ba8 100644 --- a/src/autotrain/trainers/dreambooth/datasets.py +++ b/src/autotrain/trainers/dreambooth/datasets.py @@ -96,7 +96,7 @@ def __getitem__(self, index): class DreamBoothDataset(Dataset): """ A dataset to prepare the instance and class images with the prompts for fine-tuning the model. - It pre-processes the images and the tokenizes prompts. + It pre-processes the images and tokenizes prompts. """ def __init__(self, config, tokenizers, encoder_hidden_states, instance_prompt_encoder_hidden_states): diff --git a/src/autotrain/trainers/image_classification/utils.py b/src/autotrain/trainers/image_classification/utils.py index 6b47d97045..07884f8d62 100644 --- a/src/autotrain/trainers/image_classification/utils.py +++ b/src/autotrain/trainers/image_classification/utils.py @@ -48,7 +48,7 @@ - Problem type: Image Classification -## Validation Metricsg +## Validation Metrics {validation_metrics} """