mirror of
https://git.datalinker.icu/ltdrdata/ComfyUI-Manager
synced 2025-12-09 22:24:23 +08:00
add seecoder models
ensure directory before save model file
This commit is contained in:
parent
8be5895f99
commit
b363226f2b
@ -16,7 +16,7 @@ sys.path.append('../..')
|
||||
from torchvision.datasets.utils import download_url
|
||||
|
||||
# ensure .js
|
||||
print("### Loading: ComfyUI-Manager (V0.6.4)")
|
||||
print("### Loading: ComfyUI-Manager (V0.6.5)")
|
||||
|
||||
comfy_path = os.path.dirname(folder_paths.__file__)
|
||||
custom_nodes_path = os.path.join(comfy_path, 'custom_nodes')
|
||||
@ -344,6 +344,9 @@ def download_url_with_agent(url, save_path):
|
||||
response = urllib.request.urlopen(req)
|
||||
data = response.read()
|
||||
|
||||
if not os.path.exists(os.path.dirname(save_path)):
|
||||
os.makedirs(os.path.dirname(save_path))
|
||||
|
||||
with open(save_path, 'wb') as f:
|
||||
f.write(data)
|
||||
|
||||
|
||||
@ -479,6 +479,36 @@
|
||||
"reference": "https://github.com/facebookresearch/segment-anything#model-checkpoints",
|
||||
"filename": "sam_vit_b_01ec64.pth",
|
||||
"url": "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth"
|
||||
},
|
||||
{
|
||||
"name": "seecoder v1.0",
|
||||
"type": "seecoder",
|
||||
"base": "SEECODER",
|
||||
"save_path": "seecoders",
|
||||
"description": "SeeCoder model",
|
||||
"reference": "https://huggingface.co/shi-labs/prompt-free-diffusion/tree/main/pretrained/pfd/seecoder",
|
||||
"filename": "seecoder-v1-0.safetensors",
|
||||
"url": "https://huggingface.co/shi-labs/prompt-free-diffusion/resolve/main/pretrained/pfd/seecoder/seecoder-v1-0.safetensors"
|
||||
},
|
||||
{
|
||||
"name": "seecoder pa v1.0",
|
||||
"type": "seecoder",
|
||||
"base": "SEECODER",
|
||||
"save_path": "seecoders",
|
||||
"description": "SeeCoder model",
|
||||
"reference": "https://huggingface.co/shi-labs/prompt-free-diffusion/tree/main/pretrained/pfd/seecoder",
|
||||
"filename": "seecoder-pa-v1-0.safetensors",
|
||||
"url": "https://huggingface.co/shi-labs/prompt-free-diffusion/resolve/main/pretrained/pfd/seecoder/seecoder-pa-v1-0.safetensors"
|
||||
},
|
||||
{
|
||||
"name": "seecoder anime v1.0",
|
||||
"type": "seecoder",
|
||||
"base": "SEECODER",
|
||||
"save_path": "seecoders",
|
||||
"description": "SeeCoder model",
|
||||
"reference": "https://huggingface.co/shi-labs/prompt-free-diffusion/tree/main/pretrained/pfd/seecoder",
|
||||
"filename": "seecoder-anime-v1-0.safetensors",
|
||||
"url": "https://huggingface.co/shi-labs/prompt-free-diffusion/resolve/main/pretrained/pfd/seecoder/seecoder-anime-v1-0.safetensors"
|
||||
}
|
||||
]
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user