diff --git a/hellomeme/utils.py b/hellomeme/utils.py index dc79371..f0b2196 100644 --- a/hellomeme/utils.py +++ b/hellomeme/utils.py @@ -41,8 +41,9 @@ def get_torch_device(gpu_id=-1): return torch.device(f"cuda:{gpu_id}") def load_face_toolkits(dtype=torch.float16, gpu_id=-1, modelscope=False): - pd_fpg_motion = creat_model_from_cloud(FanEncoder, 'songkey/pd_fgc_motion') - image_encoder = creat_model_from_cloud(CLIPVisionModelWithProjection, 'songkey/IP-Adapter', subfolder='models/image_encoder') + pd_fpg_motion = creat_model_from_cloud(FanEncoder, 'songkey/pd_fgc_motion', modelscope=modelscope) + image_encoder = creat_model_from_cloud(CLIPVisionModelWithProjection, 'songkey/IP-Adapter', + subfolder='models/image_encoder', modelscope=modelscope) image_encoder.to(dtype=dtype).cpu() pd_fpg_motion.to(dtype=dtype).cpu() diff --git a/pyproject.toml b/pyproject.toml index bcf1417..6fd20b0 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.26" +version = "1.3.27" license = {file = "LICENSE"} [project.urls]