Skip to content

Commit ac8ba00

Browse files
committed
change huggingface log level
1 parent fcebc95 commit ac8ba00

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

models/Orion-14B-Chat-Int4/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,8 @@ async def load(self) -> bool:
1212
import torch
1313
from transformers import AutoModelForCausalLM, AutoTokenizer
1414
from transformers.generation.utils import GenerationConfig
15-
from transformers.utils.logging import disable_progress_bar
16-
disable_progress_bar()
15+
from transformers.utils.logging import disable_progress_bar, set_verbosity_debug
16+
disable_progress_bar(); set_verbosity_debug()
1717
self.tokenizer = AutoTokenizer.from_pretrained(self.MODEL_NAME, use_fast=False, trust_remote_code=True)
1818
self.model = AutoModelForCausalLM.from_pretrained(self.MODEL_NAME, device_map="auto", torch_dtype=torch.float16, trust_remote_code=True)
1919
self.model.generation_config = GenerationConfig.from_pretrained(self.MODEL_NAME)

0 commit comments

Comments
 (0)