fixed a bug
This commit is contained in:
+3
-2
@@ -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()
|
||||
|
||||
+1
-1
@@ -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]
|
||||
|
||||
Reference in New Issue
Block a user