lc

hellomeme/tools/utils.py

@@ -144,8 +144,9 @@ def creat_model_from_cloud(model_cls,
                             cache_dir=None,
                             subfolder=None,
                             hf_token=None):
-
-    if osp.isfile(model_id) and model_id.endswith('.safetensors'):
+    if osp.isdir(model_id):
+        model = model_cls.from_pretrained(model_id)
+    elif osp.isfile(model_id) and model_id.endswith('.safetensors'):
         model = model_cls.from_single_file(model_id)
     else:
         if modelscope:

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.27"
+version = "1.3.28"
 license = {file = "LICENSE"}
 
 [project.urls]