Firas HADJ KACEM commited on
Commit
2ce05e0
·
1 Parent(s): 7b8a9fa

keep only lightweight models

Browse files
Files changed (1) hide show
  1. 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
- "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
 
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