From bd158fc62ac72fb7b6673894ebc5a37db8adfcb2 Mon Sep 17 00:00:00 2001
From: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Date: Tue, 3 Jun 2025 17:42:28 +0800
Subject: [PATCH] docs: fix PyTorch compression command

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
---
 doc/freeze/compress.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/doc/freeze/compress.md b/doc/freeze/compress.md
index f7b5a39aa3..d827c71525 100644
--- a/doc/freeze/compress.md
+++ b/doc/freeze/compress.md
@@ -77,7 +77,7 @@ dp compress -i graph.pb -o graph-compress.pb
 :::{tab-item} PyTorch {{ pytorch_icon }}
 
 ```bash
-dp compress -i model.pth -o model-compress.pth
+dp --pt compress -i model.pth -o model-compress.pth
 ```
 
 :::