From 1c8a368b4b011e4409d9fc0c9205a1fd86929fc7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?M=2E=20Tolga=20Cang=C3=B6z?= Date: Mon, 26 Feb 2024 16:58:45 +0300 Subject: [PATCH] Fix typos in comments and docstrings --- src/autotrain/trainers/dreambooth/__main__.py | 2 +- src/autotrain/trainers/dreambooth/datasets.py | 2 +- src/autotrain/trainers/image_classification/utils.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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} """