minor changes

kscalelabs · Jul 24, 2024 · c2b45b6 · c2b45b6
1 parent fe45b23
commit c2b45b6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -2,7 +2,9 @@
 
 Minimal implementation of consistency models in PyTorch.
 
-This repository was the code created as a part of Nathan's talk on Flow Matching, which also leaked a bit into Consistency Modeling, and useful optimization for flow matching that reduces the number of sampling steps required in order to reach final model prediction
+This repository was the code created as a part of Nathan's talk on Flow Matching, which also leaked a bit into Consistency Modeling, a useful optimization for flow matching that reduces the number of sampling steps required in order to reach final model prediction.
+
+The key idea is moving from noisy images to non-noisy images by following a flow "path" between Gaussian noise and the target image!
 
 ## Getting Started
 

diff --git a/mypy.ini b/mypy.ini
@@ -3,6 +3,8 @@
 python_version = 3.8
 warn_return_any = True
 warn_unused_configs = True
+install_types = True
+non_interactive = True
 
 # Ignore missing imports for torchvision
 [mypy-torchvision.*]