mirror of
https://github.com/Comfy-Org/ComfyUI-Manager.git
synced 2025-12-17 02:12:58 +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
|
from torchvision.datasets.utils import download_url
|
||||||
|
|
||||||
# ensure .js
|
# ensure .js
|
||||||
print("### Loading: ComfyUI-Manager (V0.6.4)")
|
print("### Loading: ComfyUI-Manager (V0.6.5)")
|
||||||
|
|
||||||
comfy_path = os.path.dirname(folder_paths.__file__)
|
comfy_path = os.path.dirname(folder_paths.__file__)
|
||||||
custom_nodes_path = os.path.join(comfy_path, 'custom_nodes')
|
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)
|
response = urllib.request.urlopen(req)
|
||||||
data = response.read()
|
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:
|
with open(save_path, 'wb') as f:
|
||||||
f.write(data)
|
f.write(data)
|
||||||
|
|
||||||
|
|||||||
@ -479,6 +479,36 @@
|
|||||||
"reference": "https://github.com/facebookresearch/segment-anything#model-checkpoints",
|
"reference": "https://github.com/facebookresearch/segment-anything#model-checkpoints",
|
||||||
"filename": "sam_vit_b_01ec64.pth",
|
"filename": "sam_vit_b_01ec64.pth",
|
||||||
"url": "https://dl.fbaipublicfiles.com/segment_anything/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…
Reference in New Issue
Block a user