nodes/HelloVision-ComfyUI_HelloMeme
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixed a bug

Browse Source
This commit is contained in:
songkey-win
2025-06-27 16:36:06 +08:00
parent 434440fb89
commit 6e19fc3df6
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hellomeme/utils.py
+3 -2
View File
@@ -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()