feat(ml): debug for canny threshold

jolibrain · Oct 5, 2024 · dca0bfa · dca0bfa
1 parent 84f718a
commit dca0bfa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models/palette_model.py b/models/palette_model.py
@@ -377,8 +377,8 @@ def set_input(self, data):
                         batch_cond_image = fill_img_with_random_sketch(
                             image.unsqueeze(0),
                             mask.unsqueeze(0),
-                            low_threshold_random=low,
-                            high_threshold_random=high,
+                            cur_low_threshold_random=low,
+                            cur_high_threshold_random=high,
                         ).squeeze(0)
                     elif "sam" in fill_img_with_random_sketch.__name__:
                         batch_cond_image = fill_img_with_random_sketch(