diff --git a/example_workflows/image_generation.json b/example_workflows/image_generation.json index ba9ea46..5f05381 100644 --- a/example_workflows/image_generation.json +++ b/example_workflows/image_generation.json @@ -514,7 +514,7 @@ "[preset]DisneyPixarCartoonB", "None", "same as checkpoint", - "v5b", + "v5c", "x1", "huggingface", 1, diff --git a/example_workflows/image_style_transfer.json b/example_workflows/image_style_transfer.json index 531ffd9..89f396d 100644 --- a/example_workflows/image_style_transfer.json +++ b/example_workflows/image_style_transfer.json @@ -211,7 +211,7 @@ "[preset]animeanything_v10", "[preset]ghibli_anime_v1", "same as checkpoint", - "v5b", + "v5c", "x1", "huggingface", 1.2, diff --git a/hellomeme/tools/utils.py b/hellomeme/tools/utils.py index ecd88ee..6306398 100644 --- a/hellomeme/tools/utils.py +++ b/hellomeme/tools/utils.py @@ -156,10 +156,16 @@ def creat_model_from_cloud(model_cls, print(e) assert False, "@@ Failed to download model from modelscope (using `hugginface`)" - model = model_cls.from_pretrained(model_path, subfolder=subfolder) + if subfolder is None: + model = model_cls.from_pretrained(model_path) + else: + model = model_cls.from_pretrained(model_path, subfolder=subfolder) else: try: - model = model_cls.from_pretrained(model_id, subfolder=subfolder, cache_dir=cache_dir, token=hf_token) + if subfolder is None: + model = model_cls.from_pretrained(model_id, cache_dir=cache_dir, token=hf_token) + else: + model = model_cls.from_pretrained(model_id, subfolder=subfolder, cache_dir=cache_dir, token=hf_token) except Exception as e: print(e) assert False, "@@ `huggingface-cli login` or using `modelscope`" diff --git a/pyproject.toml b/pyproject.toml index 5e5b2ae..079e544 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "comfyui_hellomeme" description = "This repository is the official implementation of the [a/HelloMeme](https://arxiv.org/pdf/2410.22901) ComfyUI interface" -version = "1.3.24" +version = "1.3.25" license = {file = "LICENSE"} [project.urls]