| import subprocess | |
| import os | |
| base_url = "https://huggingface.co/nolanaatama/models/resolve/main/" | |
| output_directory = "/content/imagination/extensions/microsoftexcel-controlnet/models/" | |
| files_to_download = [ | |
| "control_v11e_sd15_ip2p_fp16.safetensors", | |
| "control_v11e_sd15_shuffle_fp16.safetensors", | |
| "control_v11p_sd15_canny_fp16.safetensors", | |
| "control_v11p_sd15_depth_fp16.safetensors", | |
| "control_v11p_sd15_inpaint_fp16.safetensors", | |
| "control_v11p_sd15_lineart_fp16.safetensors", | |
| "control_v11p_sd15_mlsd_fp16.safetensors", | |
| "control_v11p_sd15_normalbae_fp16.safetensors", | |
| "control_v11p_sd15_openpose_fp16.safetensors", | |
| "control_v11p_sd15_scribble_fp16.safetensors", | |
| "control_v11p_sd15_seg_fp16.safetensors", | |
| "control_v11p_sd15_softedge_fp16.safetensors", | |
| "control_v11p_sd15s2_lineart_anime_fp16.safetensors", | |
| "control_v11f1e_sd15_tile_fp16.safetensors", | |
| "t2iadapter_canny_sd14v1.pth", | |
| "t2iadapter_color_sd14v1.pth", | |
| "t2iadapter_depth_sd14v1.pth", | |
| "t2iadapter_keypose_sd14v1.pth", | |
| "t2iadapter_openpose_sd14v1.pth", | |
| "t2iadapter_seg_sd14v1.pth", | |
| "t2iadapter_sketch_sd14v1.pth", | |
| "t2iadapter_style_sd14v1.pth" | |
| ] | |
| os.makedirs(output_directory, exist_ok=True) | |
| for file_name in files_to_download: | |
| file_url = base_url + file_name | |
| save_path = os.path.join(output_directory, file_name) | |
| subprocess.run(["wget", file_url, "-O", save_path]) | |
| print(f"Downloaded {file_name}") | |
| print("All files downloaded successfully!") | |