Update meme.py

2024-11-07 20:59:36 +08:00
parent 72d8a959a6
commit 0040da70b4
1 changed files with 2 additions and 2 deletions
@@ -20,7 +20,7 @@ cur_dir = osp.dirname(osp.abspath(__file__))
 installed_packages = [package.name for package in importlib.metadata.distributions()]

 REQUIRED = {
-  'diffusers', 'transformers', 'einops', 'opencv-python', 'tqdm', 'pillow', 'onnxruntime', 'onnx', 'safetensors', 'accelerate', 'peft'
+  'diffusers', 'transformers', 'einops', 'opencv-python', 'tqdm', 'pillow', 'onnxruntime-gpu', 'onnx', 'safetensors', 'accelerate', 'peft'
 }

 missing = [name for name in REQUIRED if name not in installed_packages]
@@ -414,4 +414,4 @@ NODE_DISPLAY_NAME_MAPPINGS = {
    "HMPipelineImage": "HelloMeme Image Pipeline",
    "HMPipelineVideo": "HelloMeme Video Pipeline",
    "CropReferenceImage": "Crop Reference Image",
-}
+}