diff --git a/setup.py b/setup.py
index 14465ba6..b8f8cac0 100644
--- a/setup.py
+++ b/setup.py
@@ -1,5 +1,6 @@
 import os
 import torch
+import platform
 from pathlib import Path
 from setuptools import setup, find_packages
 
@@ -45,6 +46,10 @@
     "huggingface_hub>=0.26.5",
 ]
 
+if 'riscv' in platform.machine().lower():
+    # remove triton for riscv platform
+    requirements.remove("triton")
+
 setup(
     packages=find_packages(),
     install_requires=requirements,