Update app.py

improve sketch, canny, depth

app.py

@@ -30,7 +30,9 @@ urls = {
         'models/t2iadapter_openpose_sd14v1.pth', 'models/t2iadapter_seg_sd14v1.pth',
         'models/t2iadapter_sketch_sd14v1.pth', 'models/t2iadapter_depth_sd14v1.pth',
         'third-party-models/body_pose_model.pth', "models/t2iadapter_style_sd14v1.pth",
-        "models/t2iadapter_canny_sd14v1.pth", 'third-party-models/table5_pidinet.pth'
+        "models/t2iadapter_canny_sd14v1.pth", 'third-party-models/table5_pidinet.pth',
+        "models/t2iadapter_canny_sd15v2.pth", "models/t2iadapter_depth_sd15v2.pth",
+        "models/t2iadapter_sketch_sd15v2.pth"
     ],
     'runwayml/stable-diffusion-v1-5': ['v1-5-pruned-emaonly.ckpt'],
     'andite/anything-v4.0': ['anything-v4.0-pruned.ckpt', 'anything-v4.0.vae.pt'],
@@ -81,7 +83,10 @@ parser.add_argument(
 global_opt = parser.parse_args()
 global_opt.config = 'configs/stable-diffusion/sd-v1-inference.yaml'
 for cond_name in supported_cond:
-    setattr(global_opt, f'{cond_name}_adapter_ckpt', f'models/t2iadapter_{cond_name}_sd14v1.pth')
+    if cond_name in ['sketch', 'depth', 'canny']:
+        setattr(global_opt, f'{cond_name}_adapter_ckpt', f'models/t2iadapter_{cond_name}_sd15v2.pth')
+    else:
+        setattr(global_opt, f'{cond_name}_adapter_ckpt', f'models/t2iadapter_{cond_name}_sd14v1.pth')
 global_opt.device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 global_opt.max_resolution = 512 * 512
 global_opt.sampler = 'ddim'