From 19867d5bb74b0e4fd6f6a5ce3fbff159c9d07b54 Mon Sep 17 00:00:00 2001
From: Anthony Wu <462072+anthonywu@users.noreply.github.com>
Date: Wed, 9 Oct 2024 14:20:49 -0700
Subject: [PATCH] introduce pre-commit typos check

---
 .pre-commit-config.yaml                                     | 6 +++++-
 _typos.toml                                                 | 5 +++++
 src/mflux/controlnet/flux_controlnet.py                     | 4 ++--
 src/mflux/flux/flux.py                                      | 2 +-
 ...layer_norm_continous.py => ada_layer_norm_continuous.py} | 0
 src/mflux/models/transformer/transformer.py                 | 2 +-
 6 files changed, 14 insertions(+), 5 deletions(-)
 create mode 100644 _typos.toml
 rename src/mflux/models/transformer/{ada_layer_norm_continous.py => ada_layer_norm_continuous.py} (100%)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index df981c5..b91cf35 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -1,7 +1,7 @@
 repos:
   - repo: https://github.com/astral-sh/ruff-pre-commit
     # Ruff version.
-    rev: v0.6.6
+    rev: v0.6.9
     hooks:
       # Run the linter.
       - id: ruff
@@ -10,3 +10,7 @@ repos:
       # Run the formatter.
       - id: ruff-format
         types_or: [python, pyi]
+  - repo: https://github.com/crate-ci/typos
+    rev: v1.26.0
+    hooks:
+      - id: typos
diff --git a/_typos.toml b/_typos.toml
new file mode 100644
index 0000000..fb03470
--- /dev/null
+++ b/_typos.toml
@@ -0,0 +1,5 @@
+# this is the config file for typos-cli, part of "pre-commit run"
+# use "typos --dump-config" to disocver other configs to override
+
+[default]
+extend-ignore-identifiers-re = ["EmbedND", "embed_nd", "alle"]
diff --git a/src/mflux/controlnet/flux_controlnet.py b/src/mflux/controlnet/flux_controlnet.py
index 3241c30..34ff435 100644
--- a/src/mflux/controlnet/flux_controlnet.py
+++ b/src/mflux/controlnet/flux_controlnet.py
@@ -118,7 +118,7 @@ class Flux1Controlnet:
         config = RuntimeConfig(config, self.model_config)
         time_steps = tqdm(range(config.num_inference_steps))
 
-        # Embedd the controlnet reference image
+        # Embed the controlnet reference image
         control_image = ImageUtil.load_image(controlnet_image_path)
         control_image = ControlnetUtil.scale_image(config.height, config.width, control_image)
         control_image = ControlnetUtil.preprocess_canny(control_image)
@@ -135,7 +135,7 @@ class Flux1Controlnet:
             key=mx.random.key(seed)
         )  # fmt: off
 
-        # 2. Embedd the prompt
+        # 2. Embed the prompt
         t5_tokens = self.t5_tokenizer.tokenize(prompt)
         clip_tokens = self.clip_tokenizer.tokenize(prompt)
         prompt_embeds = self.t5_text_encoder.forward(t5_tokens)
diff --git a/src/mflux/flux/flux.py b/src/mflux/flux/flux.py
index 41de838..389881e 100644
--- a/src/mflux/flux/flux.py
+++ b/src/mflux/flux/flux.py
@@ -82,7 +82,7 @@ class Flux1:
             key=mx.random.key(seed)
         )  # fmt: off
 
-        # 2. Embedd the prompt
+        # 2. Embed the prompt
         t5_tokens = self.t5_tokenizer.tokenize(prompt)
         clip_tokens = self.clip_tokenizer.tokenize(prompt)
         prompt_embeds = self.t5_text_encoder.forward(t5_tokens)
diff --git a/src/mflux/models/transformer/ada_layer_norm_continous.py b/src/mflux/models/transformer/ada_layer_norm_continuous.py
similarity index 100%
rename from src/mflux/models/transformer/ada_layer_norm_continous.py
rename to src/mflux/models/transformer/ada_layer_norm_continuous.py
diff --git a/src/mflux/models/transformer/transformer.py b/src/mflux/models/transformer/transformer.py
index 387304a..9922cce 100644
--- a/src/mflux/models/transformer/transformer.py
+++ b/src/mflux/models/transformer/transformer.py
@@ -5,7 +5,7 @@ from mlx import nn
 
 from mflux.config.model_config import ModelConfig
 from mflux.config.runtime_config import RuntimeConfig
-from mflux.models.transformer.ada_layer_norm_continous import (
+from mflux.models.transformer.ada_layer_norm_continuous import (
     AdaLayerNormContinuous,
 )
 from mflux.models.transformer.embed_nd import EmbedND