xu song
commited on
Commit
•
973bde6
1
Parent(s):
f0929ee
update
Browse files- 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},
|
|
|
|
|
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,
|