From 6ccb7d2c8e40efe80d0ced8c6608935eb30309cb Mon Sep 17 00:00:00 2001
From: songkey-win <zhang.shengkai@hellogroup.com>
Date: Fri, 27 Jun 2025 17:46:54 +0800
Subject: [PATCH] lc

---
 hellomeme/tools/utils.py | 5 +++--
 pyproject.toml           | 2 +-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/hellomeme/tools/utils.py b/hellomeme/tools/utils.py
index 6306398..06862e8 100644
--- a/hellomeme/tools/utils.py
+++ b/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:
diff --git a/pyproject.toml b/pyproject.toml
index 6fd20b0..3dfd8e5 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.27"
+version = "1.3.28"
 license = {file = "LICENSE"}
 
 [project.urls]