(runpod) stable diffusion 템플릿 - 통합 설치팩

통합 설치팩 BASIC 

# ------------------------------------------------
# 다운로드할 모델의 URL
model_urls = '''
https://huggingface.co/BanKaiPls/AsianModel/resolve/main/BRAV5finalfp16.safetensors
https://huggingface.co/syaimu/7th_Layer/resolve/main/7th_anime_v3/7th_anime_v3_A.safetensors
https://huggingface.co/eseutiblog/RunPod_database/resolve/main/zemihr_v2.safetensors
https://huggingface.co/datasets/useSword/Model_Default/resolve/main/breakdomainanime_A0440.safetensors
https://huggingface.co/datasets/useSword/Model_Default/resolve/main/feiwumix_v1025D.safetensors
https://huggingface.co/datasets/useSword/Model_Default/resolve/main/gvMix_gvMixO.safetensors
https://huggingface.co/datasets/useSword/Model_Default/resolve/main/henmixReal_v40.safetensors
https://huggingface.co/datasets/useSword/Model_Default/resolve/main/newmarsmix_R11.safetensors
'''

vae_urls = '''
https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors
https://huggingface.co/datasets/useSword/VAE_Default/resolve/main/counterfeitV25VAE_v25.pt
https://huggingface.co/datasets/useSword/VAE_Default/resolve/main/color101VAE_v1.safetensors
https://huggingface.co/datasets/useSword/VAE_Default/resolve/main/difconsistencyRAWVAE_v10.pt
'''

lora_urls = '''
https://huggingface.co/eseutiblog/RunPod_database/resolve/main/aliceNikke_v30.safetensors
https://huggingface.co/eseutiblog/RunPod_database/resolve/main/cuteGirlMix4_v10.safetensors
https://huggingface.co/eseutiblog/RunPod_database/resolve/main/epiNoiseoffset_v2.safetensors
https://huggingface.co/eseutiblog/RunPod_database/resolve/main/fashionGirl_v52.safetensors
https://huggingface.co/eseutiblog/RunPod_database/resolve/main/girlfriendmixV1_v20.safetensors
https://huggingface.co/eseutiblog/RunPod_database/resolve/main/gymtastic_v11.safetensors
https://huggingface.co/eseutiblog/RunPod_database/resolve/main/shojovibe_v11.safetensors
https://huggingface.co/eseutiblog/RunPod_database/resolve/main/spiderwoman_cosplay_outfit.safetensors
https://huggingface.co/eseutiblog/RunPod_database/resolve/main/volleyballUniform_v20Lora.safetensors
https://huggingface.co/eseutiblog/RunPod_database/resolve/main/hipoly3DModelLora_v20.safetensors
https://huggingface.co/datasets/useSword/Lora_Default/resolve/main/C_jenadmy_v10b.safetensors
https://huggingface.co/datasets/useSword/Lora_Default/resolve/main/cuteGirlMix4_v10.safetensors
https://huggingface.co/datasets/useSword/Lora_Default/resolve/main/cutePerson-v2.safetensors
https://huggingface.co/datasets/useSword/Lora_Default/resolve/main/epiNoiseoffset_v2.safetensors
https://huggingface.co/datasets/useSword/Lora_Default/resolve/main/FilmG2.safetensors
https://huggingface.co/datasets/useSword/Lora_Default/resolve/main/FilmReal.safetensors
https://huggingface.co/datasets/useSword/Lora_Default/resolve/main/kor_insta_girl_v12.safetensors
https://huggingface.co/datasets/useSword/Lora_Default/resolve/main/koreandolllikenessV20_v20.safetensors
https://huggingface.co/datasets/useSword/Lora_Default/resolve/main/flat_color.pt
https://huggingface.co/datasets/useSword/Lora_Default/resolve/main/iu_V35.safetensors
https://huggingface.co/datasets/useSword/VAE_Default/resolve/main/photopediaXL_45.safetensors
'''

embeding_urls = '''
https://huggingface.co/nancho2/nancho_runpods/resolve/main/NG_DeepNegative_V1_75T.pt
https://huggingface.co/eseutiblog/RunPod_database/resolve/main/bad_prompt_version2.pt
https://huggingface.co/eseutiblog/RunPod_database/resolve/main/badhandv4.pt
https://huggingface.co/eseutiblog/RunPod_database/resolve/main/easynegative.safetensors
https://huggingface.co/eseutiblog/RunPod_database/resolve/main/pureerosface_v1.pt
https://huggingface.co/eseutiblog/RunPod_database/resolve/main/ulzzang-6500-v1.1.bin
https://huggingface.co/datasets/useSword/runpod_Negative_Embeds/resolve/main/NEGS_ART_Nebula_Magic-Style_Fantasy_and_Galaxies_by_faeia.pt
https://huggingface.co/datasets/useSword/runpod_Negative_Embeds/resolve/main/NEGS_Bad_Artist_Anime.pt
https://huggingface.co/datasets/useSword/runpod_Negative_Embeds/resolve/main/NEGS_Bad_Artist_OG.pt
https://huggingface.co/datasets/useSword/runpod_Negative_Embeds/resolve/main/NEGS_Bad_Hands.pt
https://huggingface.co/datasets/useSword/runpod_Negative_Embeds/resolve/main/NEGS_Bad_Image_v1.pt
https://huggingface.co/datasets/useSword/runpod_Negative_Embeds/resolve/main/NEGS_Bad_Image_v2.pt
https://huggingface.co/datasets/useSword/runpod_Negative_Embeds/resolve/main/NEGS_Bad_Image_v3.pt
https://huggingface.co/datasets/useSword/runpod_Negative_Embeds/resolve/main/NEGS_Bad_Image_v4.pt
https://huggingface.co/datasets/useSword/runpod_Negative_Embeds/resolve/main/NEGS_Bad_Prompt_v1.pt
https://huggingface.co/datasets/useSword/runpod_Negative_Embeds/resolve/main/NEGS_Bad_Prompt_v2.pt
https://huggingface.co/datasets/useSword/runpod_Negative_Embeds/resolve/main/NEGS_Deep_Negative_v1.75T.pt
https://huggingface.co/datasets/useSword/runpod_Negative_Embeds/resolve/main/NEGS_Low_Resolution_Fix_HighStep_by_tolerantpancake.pt
https://huggingface.co/datasets/useSword/runpod_Negative_Embeds/resolve/main/NEGS_Low_Resolution_Fix_LowStep_by_tolerantpancake.pt
https://huggingface.co/datasets/useSword/runpod_Negative_Embeds/resolve/main/NEGS_Low_Resolution_Fix_MedStep_by_tolerantpancake.pt
https://huggingface.co/datasets/useSword/runpod_Negative_Embeds/resolve/main/NEGS_Low_Resolution_Fix_VHighStep_by_tolerantpancake.pt
https://huggingface.co/datasets/useSword/runpod_Negative_Embeds/resolve/main/NEGS_Low_Resolution_Fix_VLowStep_by_tolerantpancake.pt
'''

# 설치할 확장 기능
extensions = '''
https://github.com/nonnonstop/sd-webui-3d-open-pose-editor
https://github.com/butaixianran/Stable-Diffusion-Webui-Civitai-Helper.git
https://github.com/Mikubill/sd-webui-controlnet
https://github.com/hnmr293/sd-webui-cutoff
https://github.com/DominikDoom/a1111-sd-webui-tagcomplete
https://github.com/adieyal/sd-dynamic-prompts.git
https://github.com/mcmonkeyprojects/sd-dynamic-thresholding
https://github.com/pkuliyi2015/multidiffusion-upscaler-for-automatic1111
https://github.com/vladmandic/sd-extension-system-info
https://github.com/kohya-ss/sd-webui-additional-networks
https://github.com/Bing-su/adetailer
https://github.com/NeoGraph-K/sd-webui-ddsd
https://github.com/ilian6806/stable-diffusion-webui-state
https://github.com/comfyanonymous/ComfyUI
'''
# ------------------------------------------------

import os
import json
import shutil

home_dir = os.getcwd()
repo_dir = os.path.join(home_dir, 'stable-diffusion-webui')
os.chdir(repo_dir)

# 원래 있는 sd 모델 삭제
for i in [x for x in os.listdir(home_dir) if x.endswith('.ckpt')]:
    os.remove(os.path.join(home_dir, i))

# 모델 다운로드
models_dir = os.path.join(repo_dir, 'models', 'Stable-diffusion')
for i in [x.strip() for x in model_urls.split('\n') if x]:
    ! wget {i} -P {models_dir}

# vae 다운로드
vae_dir = os.path.join(repo_dir, 'models', 'VAE')
for i in [x.strip() for x in vae_urls.split('\n') if x]:
    ! wget {i} -P {vae_dir}

# lora 다운로드
lora_dir = os.path.join(repo_dir, 'models', 'Lora')
for i in [x.strip() for x in lora_urls.split('\n') if x]:
    ! wget {i} -P {lora_dir}

# embeding 다운로드
embeding_dir = os.path.join(repo_dir, 'embeddings')
for i in [x.strip() for x in embeding_urls.split('\n') if x]:
    ! wget {i} -P {embeding_dir}
    
!pip install --upgrade ultralytics

!pip3 install -U --pre dynamicprompts

!pip install --upgrade segment_anything
!pip install --upgrade mediapipe
!pip install --upgrade pillow_lut

# 필요한 확장 모듈 설치
ext_dir = os.path.join(repo_dir, 'extensions')
os.chdir(ext_dir)
for i in [x.strip() for x in extensions.split('\n') if x]:
    ! git clone {i}
os.chdir(home_dir)