Spaces:
Sleeping
Sleeping
Firas HADJ KACEM
commited on
Commit
·
2ce05e0
1
Parent(s):
7b8a9fa
keep only lightweight models
Browse files- backend/model_manager.py +10 -10
backend/model_manager.py
CHANGED
|
@@ -21,16 +21,16 @@ class ModelManager:
|
|
| 21 |
"model_id": "mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis",
|
| 22 |
"type": "bert"
|
| 23 |
},
|
| 24 |
-
"FinMA": {
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
},
|
| 29 |
-
"FinGPT": {
|
| 30 |
-
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
}
|
| 34 |
}
|
| 35 |
|
| 36 |
# Label IDs for Llama models
|
|
|
|
| 21 |
"model_id": "mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis",
|
| 22 |
"type": "bert"
|
| 23 |
},
|
| 24 |
+
# "FinMA": {
|
| 25 |
+
# "model_id": "ChanceFocus/finma-7b-full",
|
| 26 |
+
# "tokenizer_id": "ChanceFocus/finma-7b-full",
|
| 27 |
+
# "type": "llama"
|
| 28 |
+
# },
|
| 29 |
+
# "FinGPT": {
|
| 30 |
+
# "model_id": "oliverwang15/FinGPT_v32_Llama2_Sentiment_Instruction_LoRA_FT",
|
| 31 |
+
# "tokenizer_id": "meta-llama/Llama-2-7b-chat-hf",
|
| 32 |
+
# "type": "llama"
|
| 33 |
+
# }
|
| 34 |
}
|
| 35 |
|
| 36 |
# Label IDs for Llama models
|