ControlNet_Image_Comparison

Runtime error

ysharma HF staff commited on Mar 3, 2023

Commit

21f2c0a

1 Parent(s): 789421b

upd

Files changed (1) hide show

model.py CHANGED Viewed

@@ -631,7 +631,7 @@ class Model:
             127.5).cpu().numpy().clip(0, 255).astype(np.uint8)
         results = [x_samples[i] for i in range(num_samples)]
-        return [detected_map] + results
     @torch.inference_mode()
     def process_depth(self, input_image, prompt, a_prompt, n_prompt,

             127.5).cpu().numpy().clip(0, 255).astype(np.uint8)
         results = [x_samples[i] for i in range(num_samples)]
+        return [detected_map] + results, [detected_map] + results
     @torch.inference_mode()
     def process_depth(self, input_image, prompt, a_prompt, n_prompt,