diff --git a/h2o_hf/utils_real_drop/modify_llama.py b/h2o_hf/utils_real_drop/modify_llama.py
index 8515d08..c4a2001 100644
--- a/h2o_hf/utils_real_drop/modify_llama.py
+++ b/h2o_hf/utils_real_drop/modify_llama.py
@@ -8,7 +8,7 @@
 
 import torch.nn.functional as F
 
-from transformers.models.llama.configuration_llama import LlamaConfig
+# from transformers.models.llama.configuration_llama import LlamaConfig
 from transformers.models.llama.modeling_llama import (
     LlamaAttention,
     rotate_half,