add seecoder models

ensure directory before save model file
This commit is contained in:
Dr.Lt.Data 2023-05-29 16:09:28 +09:00
parent 8be5895f99
commit b363226f2b
2 changed files with 34 additions and 1 deletions

View File

@ -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)

View File

@ -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"
}
]
}