Minor changes, add assets

2023-06-27 10:20:03 -07:00
parent 4c5322ca85
commit 8cab96dea4
5 changed files with 73 additions and 13 deletions
--- a/scripts/train.py
+++ b/scripts/train.py
@@ -1,7 +1,7 @@
 from collections import defaultdict
 import contextlib
 import os
-from absl import app, flags, logging
+from absl import app, flags
 from ml_collections import config_flags
 from accelerate import Accelerator
 from accelerate.utils import set_seed