diff --git a/setup.py b/setup.py
index b24974c8e7..7581de64b0 100644
--- a/setup.py
+++ b/setup.py
@@ -123,6 +123,10 @@ def load_dep_info():
     RELEASE = True
     sys.argv.remove("--release")
 
+if "--dlfw" in sys.argv:
+    IS_DLFW_CI = True
+    sys.argv.remove("--dlfw")
+
 if (no_ts_env_var := os.environ.get("NO_TORCHSCRIPT")) is not None:
     if no_ts_env_var == "1":
         NO_TS = True