update download button

MagicQuill/scribble_color_edit.py

@@ -24,7 +24,7 @@ class ScribbleColorEditModel():
         self.ksampler = KSampler()
         self.vae_decoder = VAEDecode()
         self.blender = BlendInpaint()
-        self.ckpt_name = "SD1.5/realisticVisionV60B1_v51VAE.safetensors"
+        self.ckpt_name = os.path.normpath("SD1.5/realisticVisionV60B1_v51VAE.safetensors")
         with torch.no_grad():
             self.model, self.clip, self.vae = self.checkpoint_loader.load_checkpoint(self.ckpt_name)
         self.load_models('SD1.5', 'float16')
@@ -33,11 +33,7 @@ class ScribbleColorEditModel():
         if base_model_version == "SD1.5":
             edge_controlnet_name = "control_v11p_sd15_scribble.safetensors"
             color_controlnet_name = "color_finetune.safetensors"
-            brushnet_name = "brushnet/random_mask_brushnet_ckpt/diffusion_pytorch_model.safetensors"
-        # elif base_model_version == "SDXL":
-        #     edge_controlnet_name = "controlnet-scribble-sdxl-1.0.safetensors"
-        #     color_controlnet_name = "colorGridControlnet_v10.safetensors"
-        #     brushnet_name = "brushnet_xl/random_mask_brushnet_ckpt_sdxl_v0/diffusion_pytorch_model.safetensors"
+            brushnet_name = os.path.normpath("brushnet/random_mask_brushnet_ckpt/diffusion_pytorch_model.safetensors")
         else:
             raise ValueError("Invalid base_model_version, not supported yet!!!: {}".format(base_model_version))
         self.edge_controlnet = self.controlnet_loader.load_controlnet(edge_controlnet_name)[0]