add missing json, fix error

plugins/accumulnator.json

@@ -0,0 +1,10 @@
+{
+    "documentation": {
+        "curve": "one of 'linear' or 'log', defines how the grad_accum value changes from begin to end"
+    },
+    "begin_epoch": 3,
+    "end_epoch": 10,
+    "begin_grad_accum": 1,
+    "end_grad_accum": 100,
+    "curve": "linear"
+}

plugins/accumulnator.py

@@ -15,6 +15,10 @@ class Accumulnator(BasePlugin):
             begin_grad_accum = config['begin_grad_accum']
             end_epoch = config['end_epoch']
             end_grad_accum = config['end_grad_accum']
+            curve = config['curve']
+            if curve != 'linear':
+                raise NotImplementedError("Only 'linear' curve is implemented for now")
+
             accums_per_epoch = {}
             for i in range(begin_epoch):
                 accums_per_epoch[i] = begin_grad_accum

train.py

@@ -159,7 +159,7 @@ def save_model(save_path, ed_state: EveryDreamTrainingState, global_step: int, s
         logging.warning("  No model to save, something likely blew up on startup, not saving")
         return
 
-    if args.ema_decay_rate != None:
+    if ed_state.unet_ema is not None or ed_state.text_encoder_ema is not None:
         pipeline_ema = StableDiffusionPipeline(
             vae=ed_state.vae,
             text_encoder=ed_state.text_encoder_ema,