Fix typos

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 = []
 

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):

src/autotrain/trainers/image_classification/utils.py

@@ -48,7 +48,7 @@
 
 - Problem type: Image Classification
 
-## Validation Metricsg
+## Validation Metrics
 {validation_metrics}
 """