added uvit+hilbert

AshishKumar4 · AshishKumar4 · commit 3c7fff09b100 · 2025-05-05T16:49:28.000Z
diff --git a/training.py b/training.py
@@ -118,6 +118,7 @@ def boolean_string(s):
                         "simple_dit", 
                         "simple_mmdit", 
                         "hierarchical_mmdit",
+                        'uvit-hilbert',
                         'simple_dit-hilbert',
                         "simple_dit-hilbert", 
                         "simple_mmdit-hilbert", 
