update llm
Browse files
app.py
CHANGED
|
@@ -10,7 +10,7 @@ import json
|
|
| 10 |
args = {
|
| 11 |
'model': 'openllama_peft',
|
| 12 |
'imagebind_ckpt_path': '../model_zoo/imagebind_ckpt',
|
| 13 |
-
'vicuna_ckpt_path': '
|
| 14 |
'delta_ckpt_path': './pretrained_ckpt/llm7b_lora32_lamm98k/pytorch_model.pt',
|
| 15 |
'stage': 1,
|
| 16 |
'max_tgt_len': 128,
|
|
|
|
| 10 |
args = {
|
| 11 |
'model': 'openllama_peft',
|
| 12 |
'imagebind_ckpt_path': '../model_zoo/imagebind_ckpt',
|
| 13 |
+
'vicuna_ckpt_path': 'openlm-research/open_llama_7b',
|
| 14 |
'delta_ckpt_path': './pretrained_ckpt/llm7b_lora32_lamm98k/pytorch_model.pt',
|
| 15 |
'stage': 1,
|
| 16 |
'max_tgt_len': 128,
|