From a9a630e3de7ce7061895cc4440030913137cf424 Mon Sep 17 00:00:00 2001 From: songkey-win Date: Fri, 20 Dec 2024 18:11:29 +0800 Subject: [PATCH] re upload hf models --- hellomeme/utils.py | 13 +++++++------ meme.py | 4 ++-- 2 files changed, 9 insertions(+), 8 deletions(-) diff --git a/hellomeme/utils.py b/hellomeme/utils.py index 84da9ba..54e8bd8 100644 --- a/hellomeme/utils.py +++ b/hellomeme/utils.py @@ -42,13 +42,14 @@ def get_torch_device(gpu_id=-1): def load_face_toolkits(dtype=torch.float16, gpu_id=-1, modelscope=False): if modelscope: from modelscope import snapshot_download - pd_fpg_motion = FanEncoder.from_pretrained(snapshot_download('songkey/pd_fgc_motion')) - image_encoder = CLIPVisionModelWithProjection.from_pretrained( - snapshot_download('songkey/IP-Adapter'), subfolder='models/image_encoder') + pd_fpg_dir = snapshot_download('songkey/pd_fgc_motion') + clip_image_dir = snapshot_download('songkey/IP-Adapter') else: - pd_fpg_motion = FanEncoder.from_pretrained("songkey/pd_fgc_motion") - image_encoder = CLIPVisionModelWithProjection.from_pretrained( - 'h94/IP-Adapter', subfolder='models/image_encoder') + pd_fpg_dir = 'songkey/pd_fgc_motion' + clip_image_dir = 'songkey/IP-Adapter' + + pd_fpg_motion = FanEncoder.from_pretrained(pd_fpg_dir) + image_encoder = CLIPVisionModelWithProjection.from_pretrained(clip_image_dir, subfolder='models/image_encoder') image_encoder.to(dtype=dtype).cpu() pd_fpg_motion.to(dtype=dtype).cpu() diff --git a/meme.py b/meme.py index 679a611..c0d5320 100644 --- a/meme.py +++ b/meme.py @@ -79,7 +79,7 @@ class HMImagePipelineLoader: CATEGORY = "hellomeme" def load_pipeline(self, checkpoint=None, lora=None, vae=None, version='v2', stylize='x1', deployment='huggingface'): dtype = torch.float16 - pipeline = HMImagePipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5") + pipeline = HMImagePipeline.from_pretrained("songkey/stable-diffusion-v1-5") pipeline.to(dtype=dtype) pipeline.caryomitosis(version=version, modelscope=deployment=='modelscope') @@ -130,7 +130,7 @@ class HMVideoPipelineLoader: def load_pipeline(self, checkpoint=None, lora=None, vae=None, version='v2', stylize='x1', deployment='huggingface'): dtype = torch.float16 - pipeline = HMVideoPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5") + pipeline = HMVideoPipeline.from_pretrained("songkey/stable-diffusion-v1-5") pipeline.to(dtype=dtype) pipeline.caryomitosis(version=version, modelscope=deployment=='modelscope')