diff --git a/meme.py b/meme.py
index ed3466a..cb0de19 100644
--- a/meme.py
+++ b/meme.py
@@ -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",
-}
\ No newline at end of file
+}