JiantaoLin
commited on
Commit
·
bbdd4d9
1
Parent(s):
e70b610
new
Browse files
pipeline/kiss3d_wrapper.py
CHANGED
@@ -159,7 +159,7 @@ def init_wrapper_from_config(config_path):
|
|
159 |
# recon_model_config = None
|
160 |
# load llm
|
161 |
llm_configs = config_.get('llm', None)
|
162 |
-
if llm_configs is not None
|
163 |
logger.info('==> Loading LLM ...')
|
164 |
llm_device = llm_configs.get('device', 'cpu')
|
165 |
llm, llm_tokenizer = load_llm_model(llm_configs['base_model'])
|
|
|
159 |
# recon_model_config = None
|
160 |
# load llm
|
161 |
llm_configs = config_.get('llm', None)
|
162 |
+
if llm_configs is not None:
|
163 |
logger.info('==> Loading LLM ...')
|
164 |
llm_device = llm_configs.get('device', 'cpu')
|
165 |
llm, llm_tokenizer = load_llm_model(llm_configs['base_model'])
|