diff --git a/main_sd3.py b/main_sd3.py
index b6b8114..ff5ef07 100644
--- a/main_sd3.py
+++ b/main_sd3.py
@@ -26,6 +26,7 @@ todos:
     - [x] wandb log loss
     - [x] loss go down
 [] - Save checkpoint during and after training
+[] - inference + visualize examples
 """
 import math
 import copy