(#940) fix tensorboard

bghira · Sep 6, 2024 · d1e25e0 · d1e25e0
1 parent fa2a184
commit d1e25e0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/helpers/training/trainer.py b/helpers/training/trainer.py
@@ -1326,6 +1326,7 @@ def init_trackers(self):
             delattr(public_args, "process_group_kwargs")
             delattr(public_args, "weight_dtype")
             delattr(public_args, "base_weight_dtype")
+            delattr(public_args, "vae_kwargs")
 
             # Hash the contents of public_args to reflect a deterministic ID for a single set of params:
             public_args_hash = hashlib.md5(