소스 검색

Update utils_llama.py

Allen 1 년 전
부모
커밋
53b917b993
1개의 변경된 파일3개의 추가작업 그리고 0개의 파일을 삭제
  1. 3 0
      research/long-context-llama/H2O/utils_llama.py

+ 3 - 0
research/long-context-llama/H2O/utils_llama.py

@@ -19,6 +19,9 @@ from transformers.models.llama.modeling_llama import (
     LlamaForCausalLM,
 )
 from cache_utils import Cache
+from transformers.utils import logging
+
+logger = logging.get_logger(__name__)
 
 __all__ = ["H2OLlamaForCausalLM"]