xu song commited on
Commit
973bde6
1 Parent(s): f0929ee
Files changed (1) hide show
  1. models/cpp_qwen2.py +5 -1
models/cpp_qwen2.py CHANGED
@@ -24,6 +24,8 @@ python convert_hf_to_gguf.py /workspace/xusong/huggingface/models/Qwen1.5-0.5B-C
24
 
25
  import json
26
  import copy
 
 
27
  from models.base_model import Simulator
28
  import llama_cpp
29
  # import llama_cpp.llama_tokenizer
@@ -57,7 +59,9 @@ class Qwen2Simulator(Simulator):
57
  use_mlock=True,
58
  verbose=False,
59
  )
60
- logger.info(f"llm has been initialized: {self.llm}, n_threads={self.llm.n_threads}, n_ctx={self.llm.n_ctx}")
 
 
61
 
62
  self.generation_kwargs = dict(
63
  temperature=config.DEFAULT_TEMPERATURE,
 
24
 
25
  import json
26
  import copy
27
+ import os
28
+
29
  from models.base_model import Simulator
30
  import llama_cpp
31
  # import llama_cpp.llama_tokenizer
 
59
  use_mlock=True,
60
  verbose=False,
61
  )
62
+ logger.info(f"llm has been initialized: {self.llm}, "
63
+ f"n_threads={self.llm.n_threads}, n_ctx={self.llm.n_ctx}, "
64
+ f"env[CACHE]={os.environ.get('CACHE', None)}")
65
 
66
  self.generation_kwargs = dict(
67
  temperature=config.DEFAULT_TEMPERATURE,