Update modify_llama.py

FMInference · Dec 29, 2023 · 96c5ad0 · 96c5ad0
1 parent 81b49a1
commit 96c5ad0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,