Update MagicQuill/scribble_color_edit.py

MagicQuill/scribble_color_edit.py

@@ -41,7 +41,8 @@ class ScribbleColorEditModel():
         self.brushnet_loader.inpaint_files = get_files_with_extension('inpaint')
         print("self.brushnet_loader.inpaint_files: ", get_files_with_extension('inpaint'))
         self.brushnet = self.brushnet_loader.brushnet_loading(brushnet_name, dtype)[0]
-    
+
+    @torch.inference_mode()
     def process(self, ckpt_name, image, colored_image, positive_prompt, negative_prompt, mask, add_mask, remove_mask, grow_size, stroke_as_edge, fine_edge, edge_strength, color_strength, inpaint_strength, seed, steps, cfg, sampler_name, scheduler, base_model_version='SD1.5', dtype='float16', palette_resolution=2048):
         if ckpt_name != self.ckpt_name:
             self.ckpt_name = ckpt_name