From 2b3d8a1b63be6f0098ba157c9afa1dba25c15d33 Mon Sep 17 00:00:00 2001 From: songkey-win Date: Thu, 26 Jun 2025 23:56:21 +0800 Subject: [PATCH] deps --- environment.yml | Bin 5132 -> 0 bytes hellomeme/tools/utils.py | 4 +--- meme.py | 26 +------------------------- pyproject.toml | 2 +- requirements.txt | Bin 0 -> 630 bytes 5 files changed, 3 insertions(+), 29 deletions(-) delete mode 100644 environment.yml create mode 100644 requirements.txt diff --git a/environment.yml b/environment.yml deleted file mode 100644 index 65055a10df6b1b84c30fa81a3b837742b3dd7cd0..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 5132 zcmb7|*={095JmHOr1^(_Z!t39btInV8~V+X#bz7ZET+KQ$Ir|?c`IE-W#X1q!r&s- z8#f}O`0w9U7>8w;hrh#*aIg0#UF&#0gV|~WQNZKcQoyyvkUVOslSl3lJ(>2t+8_Be@c#;i!Tu6E-J@lzar3%lou!uEC_+e=n zBk9)j5c^$tl+>l9&!canej-Hrx--{B^oKexLKN1xO2i6l=CR6y__tf97v%>tsLPF* zvlnWx2bQ-j5%W;doCpi7Ugo6D4B5FqZYdy2A>EG1D zB4G;dV?Ja>?08{9pw;y=mj*d+TC!kOl6+H^%xSuS8FHsw)^Q@>B{6G|`C!^pv9WZS zmX3(2Lndbh`CB2xv`VtnKaVJ}33L+v)Ttk*VRYWK>KnVpa<<8v=NJH-u`wz7-7 zSy19JbT-bnW%BVgTJ7E%sWF6R-NOd+-jDmGE>oC+{}rWfy6 zppV?ET?9dn?c+(BX&=BQTHZ!d?ei=jBJL0J!RepSAkVHB^4n@I&E(@%9(&Sc#-+{z zM`lCrBWZHlSF%E{I5zMo_3Tn!_VUKLmRMQh&QkGM=68952XU~mU=zQNxSJ zk;h>Ex|d7UTPaR#C%zFS-I?RZj9bI zIZl2i^=n;Cq|5Vjqs-Q_xvnQQ5zt42*im_%;OJiRctF-$VTspNovggcO@(`VHWCf2 zEV0I#hx4sG9gw_>^8hame*GAI&K~5M&dd&`32Yw5JU|j3dF|0jZ0W^}m?muu$c3jF zvPm0L*>_`SDGNN4@r=dfXsT*VN<1MBw$U`Vj`8VT5wrI+6>|N%JOK$NxVzXh0-JEU zv?n3|thm`J+i3I5R_bX^9Ckh;_MQMrIl}ZZI|6b<%1M|@rP-LDot_un?2JbR0pY41D@|nHLZlr;Y2^S0kZpfLupPQGb=}f z`SU36MDFK+rc#~~`6NOIjz3DL=^lOuDB+Ft_LGg~D{?<|Z?9Z=67Mpo(cu@AX2<-l z_KPBH^8!Ys3H7!AXL&U%;OqI3GjXTPU}B{nB1`t8#C7yekDR$X)sh`-`&sqlwM5cP z?MdS;as`iOyQ_sS&%qqYk9u-?TWe-bKIgpIc-Ql625)2DwY%4O&Ws^zerNIg@I1;{ z_NvGyirG&vFy3(=onnxruntime.InferenceSession: start = time.perf_counter() - onnx_model = onnx.load(onnx_path) - onnx.checker.check_model(onnx_model) providers = [ ('CUDAExecutionProvider', { 'device_id': int(gpu_id), diff --git a/meme.py b/meme.py index 00ff698..743bf06 100644 --- a/meme.py +++ b/meme.py @@ -1,39 +1,14 @@ import json -import os import os.path as osp -import sys -import importlib.metadata - -cur_dir = osp.dirname(osp.abspath(__file__)) - -installed_packages = [package.name for package in importlib.metadata.distributions()] - -REQUIRED = { - 'diffusers':'0.33.1', 'transformers':'4.46.3', 'einops':'0.8.0', 'opencv-python':'4.10.0.84', 'tqdm':'4.67.0', - 'pillow':'10.2.0', 'onnxruntime-gpu':'1.18.1', 'onnx':'1.17.0', 'safetensors':'0.4.5', - 'accelerate':'1.1.1', 'peft':'0.13.2', 'modelscope':'', 'huggingface-hub':'' -} - -missing = [name for name in REQUIRED.keys() if name not in installed_packages] -missing_params = ' '.join([f'{k}' for k in missing]) -print("missing pkgs", missing_params) - -if missing: - python = sys.executable - os.system(f'{sys.executable} -m pip install {missing_params}') - # subprocess.check_call([python, '-m', 'pip', 'install', missing_params], stdout=subprocess.DEVNULL) import numpy as np import cv2 - from PIL import Image - import torch from einops import rearrange import folder_paths - from .hellomeme.utils import (get_drive_expression, get_drive_expression_pd_fgc, gen_control_heatmaps, @@ -49,6 +24,7 @@ from .hellomeme import (HMImagePipeline, HMVideoPipeline, download_file_from_cloud, creat_model_from_cloud) +cur_dir = osp.dirname(osp.abspath(__file__)) config_path = osp.join(cur_dir, 'hellomeme', 'model_config.json') with open(config_path, 'r') as f: MODEL_CONFIG = json.load(f) diff --git a/pyproject.toml b/pyproject.toml index 6480227..5e5b2ae 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.23" +version = "1.3.24" license = {file = "LICENSE"} [project.urls] diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000000000000000000000000000000000000..ad1ef22fb2f8ad16ed7472588954a545a414f2ba GIT binary patch literal 630 zcmY+C>rR6}5QWcg6Cb6fTw493?@}pHA~$7Q_2t!XhAz}hCQIPVxeWXJb5KE*YUvLh z6gXe>tV9pjzSo_OYE)5YO^%x!vjvk;hBQtA&;ACKdwr4f%o%Xffn3w(3$JA; z!4$gSaMU;G6Ull<8!mC5bHd!++C&8|>LxPfImw7tOAY$G>e-~FnRhdhvtr)1-?O~| zkLwOiZ|538@BE8yak1Cq!V&!c*>vUemMyXvLoVA}<~$S08okkheCrmY_w>el2ip;@ g7UqIy%Nci;`W=}w1XnHC(*?$fj{CB#$<0mp1CusgUjP6A literal 0 HcmV?d00001