feat: support custom_node's model path

update DB
This commit is contained in:
Dr.Lt.Data 2023-08-20 11:25:53 +09:00
parent 3ac484e393
commit d514de0278
3 changed files with 54 additions and 3 deletions

View File

@ -55,7 +55,7 @@ sys.path.append('../..')
from torchvision.datasets.utils import download_url
# ensure .js
print("### Loading: ComfyUI-Manager (V0.23.3)")
print("### Loading: ComfyUI-Manager (V0.24)")
comfy_ui_required_revision = 1240
comfy_ui_revision = "Unknown"
@ -335,7 +335,14 @@ import urllib.request
def get_model_dir(data):
if data['save_path'] != 'default':
base_model = os.path.join(folder_paths.models_dir, data['save_path'])
if '..' in data['save_path'] or data['save_path'].startswith('/'):
print(f"[WARN] '{data['save_path']}' is not allowed path. So it will be saved into 'models/etc'.")
base_model = "etc"
else:
if data['save_path'].startswith("custom_nodes"):
base_model = os.path.join(comfy_path, data['save_path'])
else:
base_model = os.path.join(folder_paths.models_dir, data['save_path'])
else:
model_type = data['type']
if model_type == "checkpoints":

View File

@ -835,6 +835,7 @@
"author": "ArtVentureX",
"title": "AnimateDiff",
"reference": "https://github.com/ArtVentureX/comfyui-animatediff",
"pip": ["flash_attn"],
"files": [
"https://github.com/ArtVentureX/comfyui-animatediff"
],

View File

@ -502,7 +502,17 @@
"reference": "https://huggingface.co/stabilityai/control-lora",
"filename": "control-lora-depth-rank128.safetensors",
"url": "https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank128/control-lora-depth-rank128.safetensors"
},
},
{
"name": "stabilityai/control-lora-recolor-rank128.safetensors",
"type": "controlnet",
"base": "SDXL",
"save_path": "default",
"description": "Control-LoRA: recolor rank128",
"reference": "https://huggingface.co/stabilityai/control-lora",
"filename": "control-lora-recolor-rank128.safetensors",
"url": "https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank128/control-lora-recolor-rank128.safetensors"
},
{
"name": "stabilityai/control-lora-sketch-rank128-metadata.safetensors",
"type": "controlnet",
@ -556,6 +566,17 @@
"url": "https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank256/control-lora-sketch-rank256.safetensors"
},
{
"name": "kohya-ss/ControlNet-LLLite: SDXL Canny Anime",
"type": "controlnet",
"base": "SDXL",
"save_path": "custom_nodes/ControlNet-LLLite-ComfyUI/models",
"description": "[46.2MB] An extremely compactly designed controlnet model (a.k.a. ControlNet-LLLite). Note: The model structure is highly experimental and may be subject to change in the future.",
"reference": "https://huggingface.co/kohya-ss/controlnet-lllite",
"filename": "controllllite_v01032064e_sdxl_canny_anime.safetensors",
"url": "https://huggingface.co/kohya-ss/controlnet-lllite/resolve/main/controllllite_v01032064e_sdxl_canny_anime.safetensors"
},
{
"name": "SDXL-controlnet: OpenPose (v2)",
"type": "controlnet",
@ -586,6 +607,7 @@
"filename": "depth-zoe-xl-v1.0-controlnet.safetensors",
"url": "https://huggingface.co/SargeZT/controlnet-sd-xl-1.0-depth-16bit-zoe/resolve/main/depth-zoe-xl-v1.0-controlnet.safetensors"
},
{
"name": "ControlNet-v1-1 (ip2p; fp16)",
"type": "controlnet",
@ -905,6 +927,27 @@
"reference": "https://huggingface.co/Bingsu/adetailer/tree/main",
"filename": "deepfashion2_yolov8s-seg.pt",
"url": "https://huggingface.co/Bingsu/adetailer/resolve/main/deepfashion2_yolov8s-seg.pt"
},
{
"name": "animatediff/mmd_sd_v14.ckpt",
"type": "animatediff",
"base": "SD1.x",
"save_path": "custom_nodes/comfyui-animatediff/models",
"description": "Pressing 'install' directly downloads the model from the ArtVentureX/AnimateDiff extension node. (Note: Requires ComfyUI-Manager V0.24 or above)",
"reference": "https://huggingface.co/guoyww/animatediff",
"filename": "mm_sd_v14.ckpt",
"url": "https://huggingface.co/guoyww/animatediff/resolve/main/mm_sd_v14.ckpt"
},
{
"name": "animatediff/mm_sd_v15.ckpt",
"type": "animatediff",
"base": "SD1.x",
"save_path": "custom_nodes/comfyui-animatediff/models",
"description": "Pressing 'install' directly downloads the model from the ArtVentureX/AnimateDiff extension node. (Note: Requires ComfyUI-Manager V0.24 or above)",
"reference": "https://huggingface.co/guoyww/animatediff",
"filename": "mm_sd_v15.ckpt",
"url": "https://huggingface.co/guoyww/animatediff/resolve/main/mm_sd_v15.ckpt"
}
]
}