JiantaoLin
commited on
Commit
·
788b78a
1
Parent(s):
d346594
new
Browse files
pipeline/kiss3d_wrapper.py
CHANGED
@@ -151,7 +151,7 @@ def init_wrapper_from_config(config_path):
|
|
151 |
|
152 |
# load llm
|
153 |
llm_configs = config_.get('llm', None)
|
154 |
-
if llm_configs is not None
|
155 |
logger.info('==> Loading LLM ...')
|
156 |
llm_device = llm_configs.get('device', 'cpu')
|
157 |
llm, llm_tokenizer = load_llm_model(llm_configs['base_model'])
|
|
|
151 |
|
152 |
# load llm
|
153 |
llm_configs = config_.get('llm', None)
|
154 |
+
if llm_configs is not None:
|
155 |
logger.info('==> Loading LLM ...')
|
156 |
llm_device = llm_configs.get('device', 'cpu')
|
157 |
llm, llm_tokenizer = load_llm_model(llm_configs['base_model'])
|