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

lc

Browse Source
This commit is contained in:
songkey-win
2024-12-20 18:17:49 +08:00
parent a9a630e3de
commit 9bad2b5775
1 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
meme.py
+12 -2
View File
@@ -79,7 +79,12 @@ 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("songkey/stable-diffusion-v1-5")
if deployment == 'modelscope':
from modelscope import snapshot_download
sd1_5_dir = snapshot_download('songkey/stable-diffusion-v1-5')
else:
sd1_5_dir = "songkey/stable-diffusion-v1-5"
pipeline = HMImagePipeline.from_pretrained(sd1_5_dir)
pipeline.to(dtype=dtype)
pipeline.caryomitosis(version=version, modelscope=deployment=='modelscope')
@@ -130,7 +135,12 @@ 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("songkey/stable-diffusion-v1-5")
if deployment == 'modelscope':
from modelscope import snapshot_download
sd1_5_dir = snapshot_download('songkey/stable-diffusion-v1-5')
else:
sd1_5_dir = "songkey/stable-diffusion-v1-5"
pipeline = HMVideoPipeline.from_pretrained(sd1_5_dir)
pipeline.to(dtype=dtype)
pipeline.caryomitosis(version=version, modelscope=deployment=='modelscope')