Fix typos and bugs (#1963)

Fixes some typo in `scripts/train_controlnet.py` and
`scripts/diff_model_train.py`.

---------

Signed-off-by: MaybeRichard <[email protected]>

Author: MaybeRichard

generation/maisi/scripts/diff_model_train.py

@@ -480,13 +480,13 @@ def diff_model_train(
     parser.add_argument(
         "--env_config",
         type=str,
-        default="./configs/environment_maisi_diff_model_train.json",
+        default="./configs/environment_maisi_diff_model.json",
         help="Path to environment configuration file",
     )
     parser.add_argument(
         "--model_config",
         type=str,
-        default="./configs/config_maisi_diff_model_train.json",
+        default="./configs/config_maisi_diff_model.json",
         help="Path to model training/inference configuration",
     )
     parser.add_argument(

generation/maisi/scripts/train_controlnet.py

@@ -43,7 +43,7 @@ def main():
     parser.add_argument(
         "-c",
         "--config-file",
-        default="./configs/config_maisi.json",
+        default="./configs/config_maisi-ddpm.json",
         help="config json file that stores network hyper-parameters",
     )
     parser.add_argument(
@@ -269,7 +269,7 @@ def main():
                 for label in weighted_loss_label:
                     roi[interpolate_label == label] = 1
                 weights[roi.repeat(1, images.shape[1], 1, 1, 1) == 1] = weighted_loss
-                loss = (F.l1_loss(noise_pred.float(), model_gt.float(), reduction="none") * weights).mean()
+                loss = (F.l1_loss(model_output.float(), model_gt.float(), reduction="none") * weights).mean()
             else:
                 loss = F.l1_loss(model_output.float(), model_gt.float())