Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ if HF_TOKEN:
|
|
32 |
|
33 |
# === Available Models for Selection ===
|
34 |
model_map = {
|
35 |
-
"FinGPT LoRA"
|
36 |
"InvestLM (AWQ)": {"id": "yixuantt/InvestLM-mistral-AWQ", "local": False},
|
37 |
"FinLLaMA (LLaMA3.1-8B)": {"id": "us4/fin-llama3.1-8b", "local": False},
|
38 |
"FinanceConnect (13B)": {"id": "ceadar-ie/FinanceConnect-13B", "local": True},
|
|
|
32 |
|
33 |
# === Available Models for Selection ===
|
34 |
model_map = {
|
35 |
+
"FinGPT LoRA" : {"id": "FinGPT/fingpt-mt_llama2-7b_lora", "local": True, "custom_loader": load_fingpt_lora},
|
36 |
"InvestLM (AWQ)": {"id": "yixuantt/InvestLM-mistral-AWQ", "local": False},
|
37 |
"FinLLaMA (LLaMA3.1-8B)": {"id": "us4/fin-llama3.1-8b", "local": False},
|
38 |
"FinanceConnect (13B)": {"id": "ceadar-ie/FinanceConnect-13B", "local": True},
|