update

YoucanBaby · YoucanBaby · commit 77835f7422d0 · 2025-12-19T12:32:18.000Z
diff --git a/.pylintrc b/.pylintrc
@@ -449,7 +449,6 @@ disable=raw-checker-failed,
         C0116,
         C0114,
         W0621,
-        E0601,
         W0718,
         R0801
 
diff --git a/train.py b/train.py
@@ -94,6 +94,8 @@ def training_step(
         # update emb layer when using num tokenizer
         path_sample = path.sample(x_0, x_1, t)
         x_t = path_sample.x_t
+    else:
+        x_t = None
 
     # text_token_mask==1 ==> generated text token
     x_t = x_t * data_info['text_token_mask'] + x_1 * (1 - data_info['text_token_mask'])

-Original file line number
+Diff line change
         C0116,
         C0114,
         W0621,
 -        E0601,
         W0718,
         R0801