trigger input optional

2024-02-14 16:22:09 +02:00
parent 92aae58f6e
commit e39a33a177
1 changed files with 1 additions and 1 deletions
@@ -824,7 +824,7 @@ class ADMD_TrainLora:
    CATEGORY = "AD_MotionDirector"
    FUNCTION = "train"

-    def train(self, admd_pipeline, steps, trigger_input):
+    def train(self, admd_pipeline, steps, trigger_input=None):
        with torch.inference_mode(False):
            train_noise_scheduler = admd_pipeline["train_noise_scheduler"]
            train_noise_scheduler_spatial = admd_pipeline["train_noise_scheduler_spatial"]