diff --git a/examples/llava/glmedge-surgery.py b/examples/llava/glmedge-surgery.py
index 7d7dc6837..16bb915d0 100644
--- a/examples/llava/glmedge-surgery.py
+++ b/examples/llava/glmedge-surgery.py
@@ -1,7 +1,7 @@
 import argparse
 import os
 import torch
-from transformers import AutoModel, AutoTokenizer
+from transformers import AutoModel
 
 ap = argparse.ArgumentParser()
 ap.add_argument("-m", "--model", help="Path to GLM model")