generator.ipynb

Kernel: Python 3

import re, os

colabname = r"{{colabname}}"
modelurl = r"{{modelurl}}"
modelname = r"{{modelname}}"
modelvaeurl = r"{{modelvaeurl}}"
modelvaename = r"{{modelvaename}}"
modelpage = r"{{modelpage}}"
modelpagename = r"{{modelpagename}}"
thanks = r"{{thanks}}"

##################### Please edit only this block ####################

# For README.md
new_modelpage = "https://civitai.com/models/41206/coremixpure"

new_colabname = "coremixpure_webui_colab"

# If from civitai model_creator_username/model_name
new_modelpagename = "CornmeisterNL/coremixpure"

# If no one has suggested it, please change 'person' to your username or delete all the strings like new_thanks = ""
# new_thanks = "<br /> (Thanks to person for the suggestion ❤)"
new_thanks = "<br /> (Thanks to Koneko❁ུ۪۪♡ for the suggestion ❤)"

# For WebUI
new_modelname = "coremixpure_v10.safetensors"

new_modelvaename = "coremixpure_v10.vae.pt"

# If link like https://civitai.com/api/download/models/16553?type=Model&format=SafeTensor
# please use new_modelurl = "\\\"https://civitai.com/api/download/models/16553?type=Model&format=SafeTensor\\\""
new_modelurl = "https://huggingface.co/ckpt/coremixpure/resolve/main/coremixpure_v10.safetensors"

new_modelvaeurl = "https://huggingface.co/ckpt/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.ckpt"

#######################################################################

with open("readme_in", 'r') as f:
    input_text_readme_in = f.read()
input_text1_readme_in = re.sub(modelpage, new_modelpage, input_text_readme_in)
input_text2_readme_in = re.sub(modelpagename, new_modelpagename, input_text1_readme_in)
input_text3_readme_in = re.sub(thanks, new_thanks, input_text2_readme_in)
input_text4_readme_in = re.sub(colabname, new_colabname, input_text3_readme_in)
with open("readme_out", 'w') as f:
    f.write(input_text4_readme_in)

output_file_lite = os.path.join(os.path.dirname(os.getcwd()), "lite", f"{new_colabname}.ipynb")
file_lite = "lite.ipynb"
with open(file_lite, 'r') as f:
    input_text_lite = f.read()
output_text1_lite = re.sub(colabname, new_colabname, input_text_lite)
output_text2_lite = re.sub(modelurl, new_modelurl, output_text1_lite)
output_text3_lite = re.sub(modelname, new_modelname, output_text2_lite)
output_text4_lite = re.sub(modelvaeurl, new_modelvaeurl, output_text3_lite)
output_text5_lite = re.sub(modelvaename, new_modelvaename, output_text4_lite)
with open(output_file_lite, 'w') as f:
    f.write(output_text5_lite)

output_file_stable = os.path.join(os.path.dirname(os.getcwd()), "stable", f"{new_colabname}.ipynb")
file_stable = "stable.ipynb"
with open(file_stable, 'r') as f:
    input_text_stable = f.read()
output_text1_stable = re.sub(colabname, new_colabname, input_text_stable)
output_text2_stable = re.sub(modelurl, new_modelurl, output_text1_stable)
output_text3_stable = re.sub(modelname, new_modelname, output_text2_stable)
output_text4_stable = re.sub(modelvaeurl, new_modelvaeurl, output_text3_stable)
output_text5_stable = re.sub(modelvaename, new_modelvaename, output_text4_stable)
with open(output_file_stable, 'w') as f:
    f.write(output_text5_stable)

output_file_nightly = os.path.join(os.path.dirname(os.getcwd()), "nightly", f"{new_colabname}.ipynb")
file_nightly = "nightly.ipynb"
with open(file_nightly, 'r') as f:
    input_text_nightly = f.read()
output_text1_nightly = re.sub(colabname, new_colabname, input_text_nightly)
output_text2_nightly = re.sub(modelurl, new_modelurl, output_text1_nightly)
output_text3_nightly = re.sub(modelname, new_modelname, output_text2_nightly)
output_text4_nightly = re.sub(modelvaeurl, new_modelvaeurl, output_text3_nightly)
output_text5_nightly = re.sub(modelvaename, new_modelvaename, output_text4_nightly)
with open(output_file_nightly, 'w') as f:
    f.write(output_text5_nightly)