inpaint 2.5 (#631)

This commit is contained in:
lllyasviel
2023-10-10 14:44:02 -07:00
committed by GitHub
parent a960d24825
commit c24be3f886
3 changed files with 7 additions and 7 deletions
+5 -5
View File
@@ -132,7 +132,7 @@ def update_all_model_names():
def downloading_inpaint_models(v):
assert v in ['v1', 'v2']
assert v in ['v1', 'v2.5']
load_file_from_url(
url='https://huggingface.co/lllyasviel/fooocus_inpaint/resolve/main/fooocus_inpaint_head.pth',
@@ -148,13 +148,13 @@ def downloading_inpaint_models(v):
)
return os.path.join(inpaint_models_path, 'fooocus_inpaint_head.pth'), os.path.join(inpaint_models_path, 'inpaint.fooocus.patch')
if v == 'v2':
if v == 'v2.5':
load_file_from_url(
url='https://huggingface.co/lllyasviel/fooocus_inpaint/resolve/main/inpaint_v2.fooocus.patch',
url='https://huggingface.co/lllyasviel/fooocus_inpaint/resolve/main/inpaint_v25.fooocus.patch',
model_dir=inpaint_models_path,
file_name='inpaint_v2.fooocus.patch'
file_name='inpaint_v25.fooocus.patch'
)
return os.path.join(inpaint_models_path, 'fooocus_inpaint_head.pth'), os.path.join(inpaint_models_path, 'inpaint_v2.fooocus.patch')
return os.path.join(inpaint_models_path, 'fooocus_inpaint_head.pth'), os.path.join(inpaint_models_path, 'inpaint_v25.fooocus.patch')
def downloading_controlnet_canny():