imspidey commited on
Commit
1c9577d
·
1 Parent(s): 32a5fe6

Create controlnet.py

Browse files
Files changed (1) hide show
  1. controlnet.py +41 -0
controlnet.py ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import requests
2
+
3
+ base_url = "https://huggingface.co/nolanaatama/models/resolve/main/"
4
+
5
+ files_to_download = [
6
+ ("control_v11e_sd15_ip2p.safetensors", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/control_v11e_sd15_ip2p.safetensors"),
7
+ ("control_v11e_sd15_shuffle_fp16.safetensors", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/control_v11e_sd15_shuffle.safetensors"),
8
+ ("control_v11p_sd15_canny.safetensors", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/control_v11p_sd15_canny.safetensors"),
9
+ ("control_v11p_sd15_depth_fp16.safetensors", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/control_v11f1p_sd15_depth.safetensors"),
10
+ ("control_v11p_sd15_inpaint_fp16.safetensors", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/control_v11p_sd15_inpaint.safetensors"),
11
+ ("control_v11p_sd15_lineart_fp16.safetensors", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/control_v11p_sd15_lineart.safetensors"),
12
+ ("control_v11p_sd15_mlsd_fp16.safetensors", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/control_v11p_sd15_mlsd.safetensors"),
13
+ ("control_v11p_sd15_normalbae_fp16.safetensors", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/control_v11p_sd15_normalbae.safetensors"),
14
+ ("control_v11p_sd15_openpose_fp16.safetensors", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/control_v11p_sd15_openpose.safetensors"),
15
+ ("control_v11p_sd15_scribble_fp16.safetensors", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/control_v11p_sd15_scribble.safetensors"),
16
+ ("control_v11p_sd15_seg_fp16.safetensors", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/control_v11p_sd15_seg.safetensors"),
17
+ ("control_v11p_sd15_softedge_fp16.safetensors", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/control_v11p_sd15_softedge.safetensors"),
18
+ ("control_v11p_sd15s2_lineart_anime_fp16.safetensors", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/control_v11p_sd15s2_lineart_anime.safetensors"),
19
+ ("control_v11f1e_sd15_tile_fp16.safetensors", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/control_v11f1e_sd15_tile.safetensors"),
20
+ ("t2iadapter_canny_sd14v1.pth", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/t2iadapter_canny_sd14v1.pth"),
21
+ ("t2iadapter_color_sd14v1.pth", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/t2iadapter_color_sd14v1.pth"),
22
+ ("t2iadapter_depth_sd14v1.pth", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/t2iadapter_depth_sd14v1.pth"),
23
+ ("t2iadapter_keypose_sd14v1.pth", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/t2iadapter_keypose_sd14v1.pth"),
24
+ ("t2iadapter_openpose_sd14v1.pth", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/t2iadapter_openpose_sd14v1.pth"),
25
+ ("t2iadapter_seg_sd14v1.pth", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/t2iadapter_seg_sd14v1.pth"),
26
+ ("t2iadapter_sketch_sd14v1.pth", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/t2iadapter_sketch_sd14v1.pth"),
27
+ ("t2iadapter_style_sd14v1.pth", "/content/microsoftexcel/extensions/microsoftexcel-controlnet/models/t2iadapter_style_sd14v1.pth"),
28
+ ]
29
+
30
+ for file_name, save_path in files_to_download:
31
+ file_url = base_url + file_name
32
+ response = requests.get(file_url, stream=True)
33
+ if response.status_code == 200:
34
+ with open(save_path, 'wb') as file:
35
+ for chunk in response.iter_content(chunk_size=8192):
36
+ file.write(chunk)
37
+ print(f"Downloaded {file_name} to {save_path}")
38
+ else:
39
+ print(f"Failed to download {file_name}")
40
+
41
+ print("All files downloaded successfully!")