Hello, I fixed the script for Neo that Anon shared. Now NeoForge recognizes everything except VAEs.
What do I need to modify?(I never use VAEs, but my TOC requires it.)
def get_custom_model_folder():
"""
Update extra network directories with user-specified values.
"""
util.printD("Get Custom Model Folder")
if hasattr(shared.cmd_opts, 'embeddings_dir') and shared.cmd_opts.embeddings_dir and os.path.isdir(shared.cmd_opts.embeddings_dir):
folders["ti"] = shared.cmd_opts.embeddings_dir
if hasattr(shared.cmd_opts, 'hypernetwork_dir') and shared.cmd_opts.hypernetwork_dir and os.path.isdir(shared.cmd_opts.hypernetwork_dir):
folders["hyper"] = shared.cmd_opts.hypernetwork_dir
if hasattr(shared.cmd_opts, 'ckpt_dirs') and shared.cmd_opts.ckpt_dirs:
if isinstance(shared.cmd_opts.ckpt_dirs, list):
for ckpt_dir in shared.cmd_opts.ckpt_dirs:
if os.path.isdir(ckpt_dir):
folders["ckp"] = ckpt_dir
break
elif os.path.isdir(shared.cmd_opts.ckpt_dirs):
folders["ckp"] = shared.cmd_opts.ckpt_dirs
if hasattr(shared.cmd_opts, 'lora_dir') and shared.cmd_opts.lora_dir and os.path.isdir(shared.cmd_opts.lora_dir):
folders["lora"] = shared.cmd_opts.lora_dir
if hasattr(shared.cmd_opts, 'vae_dir') and shared.cmd_opts.vae_dir and os.path.isdir(shared.cmd_opts.vae_dir):
folders["vae"] = shared.cmd_opts.vae_dir
if util.get_opts("ch_dl_lyco_to_lora"):
folders["lycoris"] = folders["lora"]
return
try:
if hasattr(shared.cmd_opts, 'lyco_dir') and shared.cmd_opts.lyco_dir and os.path.isdir(shared.cmd_opts.lyco_dir):
folders["lycoris"] = shared.cmd_opts.lyco_dir
except AttributeError:
try:
if hasattr(shared.cmd_opts, 'lyco_dir_backcompat') and shared.cmd_opts.lyco_dir_backcompat and os.path.isdir(shared.cmd_opts.lyco_dir_backcompat):
folders["lycoris"] = shared.cmd_opts.lyco_dir_backcompat
except AttributeError:
return
