Spaces:
Running
Running
Commit
·
c47d7dd
1
Parent(s):
f5215c3
GPT4o, fix UM GPT sunset
Browse files- api_cost/api_cost.yaml +9 -0
- app.py +1 -1
- vouchervision/VoucherVision_Config_Builder.py +1 -1
- vouchervision/model_maps.py +71 -37
api_cost/api_cost.yaml
CHANGED
|
@@ -10,6 +10,12 @@ GPT_4:
|
|
| 10 |
# GPT_4_TURBO: ###############
|
| 11 |
# in: 0.01
|
| 12 |
# out: 0.03
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
GPT_4_TURBO_0125:
|
| 14 |
in: 0.01
|
| 15 |
out: 0.03
|
|
@@ -93,5 +99,8 @@ LOCAL_MISTRAL_7B_INSTRUCT_V02:
|
|
| 93 |
in: 0.0
|
| 94 |
out: 0.0
|
| 95 |
LOCAL_CPU_MISTRAL_7B_INSTRUCT_V02_GGUF:
|
|
|
|
|
|
|
|
|
|
| 96 |
in: 0.0
|
| 97 |
out: 0.0
|
|
|
|
| 10 |
# GPT_4_TURBO: ###############
|
| 11 |
# in: 0.01
|
| 12 |
# out: 0.03
|
| 13 |
+
GPT_4o_2024_05_13:
|
| 14 |
+
in: 0.005
|
| 15 |
+
out: 0.015
|
| 16 |
+
GPT_4_TURBO_2024_04_09:
|
| 17 |
+
in: 0.01
|
| 18 |
+
out: 0.03
|
| 19 |
GPT_4_TURBO_0125:
|
| 20 |
in: 0.01
|
| 21 |
out: 0.03
|
|
|
|
| 99 |
in: 0.0
|
| 100 |
out: 0.0
|
| 101 |
LOCAL_CPU_MISTRAL_7B_INSTRUCT_V02_GGUF:
|
| 102 |
+
in: 0.0
|
| 103 |
+
out: 0.0
|
| 104 |
+
phyloforfun/mistral-7b-instruct-v2-bnb-4bit__HLT_MICH_Angiospermae_SLTPvC_v1-0_medium_OCR-C25-L25-E50-R05:
|
| 105 |
in: 0.0
|
| 106 |
out: 0.0
|
app.py
CHANGED
|
@@ -1730,7 +1730,7 @@ Based on preliminary results, the following models perform the best. We are curr
|
|
| 1730 |
- `LOCAL Mixtral 7Bx8 Instruct`
|
| 1731 |
- `LOCAL Mixtral 7B Instruct`
|
| 1732 |
|
| 1733 |
-
Larger models (e.g., `GPT 4`, `
|
| 1734 |
|
| 1735 |
The `SLTPvA_short.yaml` prompt also seems to work better with smaller LLMs (e.g., Mistral Tiny). Alternatively, enable double OCR to help the LLM focus on the OCR text given a longer prompt.
|
| 1736 |
|
|
|
|
| 1730 |
- `LOCAL Mixtral 7Bx8 Instruct`
|
| 1731 |
- `LOCAL Mixtral 7B Instruct`
|
| 1732 |
|
| 1733 |
+
Larger models (e.g., `GPT 4`, `Gemini Pro`) do not necessarily perform better for these tasks. MistralAI models exceeded our expectations and perform extremely well. PaLM 2 text-bison@001 also seems to consistently out-perform Gemini Pro.
|
| 1734 |
|
| 1735 |
The `SLTPvA_short.yaml` prompt also seems to work better with smaller LLMs (e.g., Mistral Tiny). Alternatively, enable double OCR to help the LLM focus on the OCR text given a longer prompt.
|
| 1736 |
|
vouchervision/VoucherVision_Config_Builder.py
CHANGED
|
@@ -49,7 +49,7 @@ def build_VV_config(loaded_cfg=None):
|
|
| 49 |
|
| 50 |
check_for_illegal_filenames = False
|
| 51 |
|
| 52 |
-
LLM_version_user = 'Azure GPT
|
| 53 |
prompt_version = 'SLTPvA_long.yaml' # from ["Version 1", "Version 1 No Domain Knowledge", "Version 2"]
|
| 54 |
use_LeafMachine2_collage_images = True # Use LeafMachine2 collage images
|
| 55 |
do_create_OCR_helper_image = True
|
|
|
|
| 49 |
|
| 50 |
check_for_illegal_filenames = False
|
| 51 |
|
| 52 |
+
LLM_version_user = 'Azure GPT 4' #'Azure GPT 4 Turbo 1106-preview'
|
| 53 |
prompt_version = 'SLTPvA_long.yaml' # from ["Version 1", "Version 1 No Domain Knowledge", "Version 2"]
|
| 54 |
use_LeafMachine2_collage_images = True # Use LeafMachine2 collage images
|
| 55 |
do_create_OCR_helper_image = True
|
vouchervision/model_maps.py
CHANGED
|
@@ -3,6 +3,8 @@ class ModelMaps:
|
|
| 3 |
COLORS_EXPENSE_REPORT = {
|
| 4 |
'GPT_4': '#32CD32', # Lime Green
|
| 5 |
'GPT_3_5': '#008000', # Green
|
|
|
|
|
|
|
| 6 |
'GPT_3_5_INSTRUCT': '#3CB371', # Medium Sea Green
|
| 7 |
'GPT_4_TURBO_1106': '#228B22', # Forest Green
|
| 8 |
'GPT_4_TURBO_0125': '#228B22', # Forest Green
|
|
@@ -14,11 +16,11 @@ class ModelMaps:
|
|
| 14 |
'GEMINI_PRO': '#1E00FF', #
|
| 15 |
|
| 16 |
'AZURE_GPT_4': '#800080', # Purple
|
| 17 |
-
'AZURE_GPT_4_TURBO_1106': '#9370DB', # Medium Purple
|
| 18 |
-
'AZURE_GPT_4_TURBO_0125': '#9370DB', # Medium Purple
|
| 19 |
-
'AZURE_GPT_4_32K': '#8A2BE2', # Blue Violet
|
| 20 |
-
'AZURE_GPT_3_5_INSTRUCT': '#9400D3', # Dark Violet
|
| 21 |
-
'AZURE_GPT_3_5': '#9932CC', # Dark Orchid
|
| 22 |
|
| 23 |
'OPEN_MISTRAL_7B': '#FFA07A', # Light Salmon
|
| 24 |
'OPEN_MIXTRAL_8X7B': '#FF8C00', # Dark Orange
|
|
@@ -30,21 +32,26 @@ class ModelMaps:
|
|
| 30 |
'LOCAL_MISTRAL_7B_INSTRUCT_V02': '#4a4a4a', # Gray
|
| 31 |
|
| 32 |
'LOCAL_CPU_MISTRAL_7B_INSTRUCT_V02_GGUF': '#bababa', # Gray
|
|
|
|
|
|
|
| 33 |
}
|
| 34 |
|
| 35 |
MODELS_OPENAI = ["GPT 4",
|
| 36 |
"GPT 4 32k",
|
|
|
|
|
|
|
| 37 |
"GPT 4 Turbo 0125-preview",
|
| 38 |
"GPT 4 Turbo 1106-preview",
|
| 39 |
"GPT 3.5 Turbo",
|
| 40 |
"GPT 3.5 Instruct",
|
| 41 |
|
| 42 |
"Azure GPT 4",
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
| 46 |
-
|
| 47 |
-
|
|
|
|
| 48 |
|
| 49 |
MODELS_GOOGLE = ["PaLM 2 text-bison@001",
|
| 50 |
"PaLM 2 text-bison@002",
|
|
@@ -60,24 +67,27 @@ class ModelMaps:
|
|
| 60 |
|
| 61 |
MODELS_LOCAL = ["LOCAL Mixtral 8x7B Instruct v0.1",
|
| 62 |
"LOCAL Mistral 7B Instruct v0.2",
|
| 63 |
-
"LOCAL CPU Mistral 7B Instruct v0.2 GGUF",
|
|
|
|
| 64 |
|
| 65 |
-
MODELS_GUI_DEFAULT = "Azure GPT
|
| 66 |
|
| 67 |
version_mapping_cost = {
|
| 68 |
'GPT 4 32k': 'GPT_4_32K',
|
| 69 |
'GPT 4': 'GPT_4',
|
|
|
|
|
|
|
| 70 |
'GPT 4 Turbo 0125-preview': 'GPT_4_TURBO_0125',
|
| 71 |
'GPT 4 Turbo 1106-preview': 'GPT_4_TURBO_1106',
|
| 72 |
'GPT 3.5 Instruct': 'GPT_3_5_INSTRUCT',
|
| 73 |
'GPT 3.5 Turbo': 'GPT_3_5',
|
| 74 |
|
| 75 |
-
'Azure GPT 4 32k': 'AZURE_GPT_4_32K',
|
| 76 |
'Azure GPT 4': 'AZURE_GPT_4',
|
| 77 |
-
'Azure GPT 4 Turbo 0125-preview': 'AZURE_GPT_4_TURBO_0125',
|
| 78 |
-
'Azure GPT 4 Turbo 1106-preview': 'AZURE_GPT_4_TURBO_1106',
|
| 79 |
-
'Azure GPT 3.5 Instruct': 'AZURE_GPT_3_5_INSTRUCT',
|
| 80 |
-
'Azure GPT 3.5 Turbo': 'AZURE_GPT_3_5',
|
| 81 |
|
| 82 |
'Gemini Pro': 'GEMINI_PRO',
|
| 83 |
'PaLM 2 text-unicorn@001': 'PALM2_TU_1',
|
|
@@ -94,25 +104,29 @@ class ModelMaps:
|
|
| 94 |
'LOCAL Mistral 7B Instruct v0.2': 'LOCAL_MISTRAL_7B_INSTRUCT_V02',
|
| 95 |
|
| 96 |
'LOCAL CPU Mistral 7B Instruct v0.2 GGUF': 'LOCAL_CPU_MISTRAL_7B_INSTRUCT_V02_GGUF',
|
|
|
|
|
|
|
| 97 |
}
|
| 98 |
|
| 99 |
@classmethod
|
| 100 |
def get_version_has_key(cls, key, has_key_openai, has_key_azure_openai, has_key_google_application_credentials, has_key_mistral):
|
| 101 |
# Define the mapping for 'has_key' values
|
| 102 |
version_has_key = {
|
|
|
|
| 103 |
'GPT 4 Turbo 1106-preview': has_key_openai,
|
| 104 |
'GPT 4 Turbo 0125-preview': has_key_openai,
|
| 105 |
'GPT 4': has_key_openai,
|
|
|
|
| 106 |
'GPT 4 32k': has_key_openai,
|
| 107 |
'GPT 3.5 Turbo': has_key_openai,
|
| 108 |
'GPT 3.5 Instruct': has_key_openai,
|
| 109 |
|
| 110 |
-
'Azure GPT 3.5 Turbo': has_key_azure_openai,
|
| 111 |
-
'Azure GPT 3.5 Instruct': has_key_azure_openai,
|
| 112 |
'Azure GPT 4': has_key_azure_openai,
|
| 113 |
-
'Azure GPT 4 Turbo 1106-preview': has_key_azure_openai,
|
| 114 |
-
'Azure GPT 4 Turbo 0125-preview': has_key_azure_openai,
|
| 115 |
-
'Azure GPT 4 32k': has_key_azure_openai,
|
| 116 |
|
| 117 |
'PaLM 2 text-bison@001': has_key_google_application_credentials,
|
| 118 |
'PaLM 2 text-bison@002': has_key_google_application_credentials,
|
|
@@ -129,12 +143,16 @@ class ModelMaps:
|
|
| 129 |
'LOCAL Mistral 7B Instruct v0.2': True,
|
| 130 |
|
| 131 |
'LOCAL CPU Mistral 7B Instruct v0.2 GGUF': True,
|
|
|
|
|
|
|
| 132 |
}
|
| 133 |
return version_has_key.get(key)
|
| 134 |
|
| 135 |
@classmethod
|
| 136 |
def get_version_mapping_is_azure(cls, key):
|
| 137 |
version_mapping_is_azure = {
|
|
|
|
|
|
|
| 138 |
"GPT 4 Turbo 1106-preview": False,
|
| 139 |
"GPT 4 Turbo 0125-preview": False,
|
| 140 |
'GPT 4': False,
|
|
@@ -142,12 +160,12 @@ class ModelMaps:
|
|
| 142 |
'GPT 3.5 Turbo': False,
|
| 143 |
'GPT 3.5 Instruct': False,
|
| 144 |
|
| 145 |
-
'Azure GPT 3.5 Turbo': True,
|
| 146 |
-
'Azure GPT 3.5 Instruct': True,
|
| 147 |
'Azure GPT 4': True,
|
| 148 |
-
'Azure GPT 4 Turbo 1106-preview': True,
|
| 149 |
-
'Azure GPT 4 Turbo 0125-preview': True,
|
| 150 |
-
'Azure GPT 4 32k': True,
|
| 151 |
|
| 152 |
'PaLM 2 text-bison@001': False,
|
| 153 |
'PaLM 2 text-bison@002': False,
|
|
@@ -164,6 +182,8 @@ class ModelMaps:
|
|
| 164 |
'LOCAL Mistral 7B Instruct v0.2': False,
|
| 165 |
|
| 166 |
'LOCAL CPU Mistral 7B Instruct v0.2 GGUF': False,
|
|
|
|
|
|
|
| 167 |
}
|
| 168 |
return version_mapping_is_azure.get(key)
|
| 169 |
|
|
@@ -182,7 +202,13 @@ class ModelMaps:
|
|
| 182 |
|
| 183 |
elif key == 'GPT_4_32K':
|
| 184 |
return 'gpt-4-32k'
|
|
|
|
|
|
|
|
|
|
| 185 |
|
|
|
|
|
|
|
|
|
|
| 186 |
elif key == 'GPT_4_TURBO_1106':
|
| 187 |
return 'gpt-4-1106-preview'
|
| 188 |
|
|
@@ -190,23 +216,23 @@ class ModelMaps:
|
|
| 190 |
return 'gpt-4-0125-preview'
|
| 191 |
|
| 192 |
### Azure
|
| 193 |
-
elif key == 'AZURE_GPT_3_5':
|
| 194 |
-
|
| 195 |
|
| 196 |
-
elif key == 'AZURE_GPT_3_5_INSTRUCT':
|
| 197 |
-
|
| 198 |
|
| 199 |
elif key == 'AZURE_GPT_4':
|
| 200 |
return "gpt-4"
|
| 201 |
|
| 202 |
-
elif key == 'AZURE_GPT_4_TURBO_1106':
|
| 203 |
-
|
| 204 |
|
| 205 |
-
elif key == 'AZURE_GPT_4_TURBO_0125':
|
| 206 |
-
|
| 207 |
|
| 208 |
-
elif key == 'AZURE_GPT_4_32K':
|
| 209 |
-
|
| 210 |
|
| 211 |
### Google
|
| 212 |
elif key == 'PALM2_TB_1':
|
|
@@ -248,6 +274,14 @@ class ModelMaps:
|
|
| 248 |
### Mistral LOCAL CPU
|
| 249 |
elif key == 'LOCAL_CPU_MISTRAL_7B_INSTRUCT_V02_GGUF':
|
| 250 |
return 'Mistral-7B-Instruct-v0.2-GGUF'
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 251 |
|
| 252 |
else:
|
| 253 |
raise ValueError(f"Invalid model name {key}. See model_maps.py")
|
|
|
|
| 3 |
COLORS_EXPENSE_REPORT = {
|
| 4 |
'GPT_4': '#32CD32', # Lime Green
|
| 5 |
'GPT_3_5': '#008000', # Green
|
| 6 |
+
'GPT_4_TURBO_2024_04_09': '#32CD32', # Lime Green
|
| 7 |
+
'GPT_4o_2024_05_13': '#3CB371', # Lime Green gpt-4o-2024-05-13
|
| 8 |
'GPT_3_5_INSTRUCT': '#3CB371', # Medium Sea Green
|
| 9 |
'GPT_4_TURBO_1106': '#228B22', # Forest Green
|
| 10 |
'GPT_4_TURBO_0125': '#228B22', # Forest Green
|
|
|
|
| 16 |
'GEMINI_PRO': '#1E00FF', #
|
| 17 |
|
| 18 |
'AZURE_GPT_4': '#800080', # Purple
|
| 19 |
+
# 'AZURE_GPT_4_TURBO_1106': '#9370DB', # Medium Purple
|
| 20 |
+
# 'AZURE_GPT_4_TURBO_0125': '#9370DB', # Medium Purple
|
| 21 |
+
# 'AZURE_GPT_4_32K': '#8A2BE2', # Blue Violet
|
| 22 |
+
# 'AZURE_GPT_3_5_INSTRUCT': '#9400D3', # Dark Violet
|
| 23 |
+
# 'AZURE_GPT_3_5': '#9932CC', # Dark Orchid
|
| 24 |
|
| 25 |
'OPEN_MISTRAL_7B': '#FFA07A', # Light Salmon
|
| 26 |
'OPEN_MIXTRAL_8X7B': '#FF8C00', # Dark Orange
|
|
|
|
| 32 |
'LOCAL_MISTRAL_7B_INSTRUCT_V02': '#4a4a4a', # Gray
|
| 33 |
|
| 34 |
'LOCAL_CPU_MISTRAL_7B_INSTRUCT_V02_GGUF': '#bababa', # Gray
|
| 35 |
+
|
| 36 |
+
'phyloforfun/mistral-7b-instruct-v2-bnb-4bit__HLT_MICH_Angiospermae_SLTPvC_v1-0_medium_OCR-C25-L25-E50-R05': '#bababa', # Gray
|
| 37 |
}
|
| 38 |
|
| 39 |
MODELS_OPENAI = ["GPT 4",
|
| 40 |
"GPT 4 32k",
|
| 41 |
+
"GPT 4o 2024-05-13", #GPT_4o_2024_05_13
|
| 42 |
+
"GPT 4 Turbo 2024-04-09",#GPT_4_TURBO_2024_04_09
|
| 43 |
"GPT 4 Turbo 0125-preview",
|
| 44 |
"GPT 4 Turbo 1106-preview",
|
| 45 |
"GPT 3.5 Turbo",
|
| 46 |
"GPT 3.5 Instruct",
|
| 47 |
|
| 48 |
"Azure GPT 4",
|
| 49 |
+
# "Azure GPT 4 32k",
|
| 50 |
+
# "Azure GPT 4 Turbo 0125-preview",
|
| 51 |
+
# "Azure GPT 4 Turbo 1106-preview",
|
| 52 |
+
# "Azure GPT 3.5 Turbo",
|
| 53 |
+
# "Azure GPT 3.5 Instruct",
|
| 54 |
+
]
|
| 55 |
|
| 56 |
MODELS_GOOGLE = ["PaLM 2 text-bison@001",
|
| 57 |
"PaLM 2 text-bison@002",
|
|
|
|
| 67 |
|
| 68 |
MODELS_LOCAL = ["LOCAL Mixtral 8x7B Instruct v0.1",
|
| 69 |
"LOCAL Mistral 7B Instruct v0.2",
|
| 70 |
+
"LOCAL CPU Mistral 7B Instruct v0.2 GGUF",
|
| 71 |
+
'phyloforfun/mistral-7b-instruct-v2-bnb-4bit__HLT_MICH_Angiospermae_SLTPvC_v1-0_medium_OCR-C25-L25-E50-R05']
|
| 72 |
|
| 73 |
+
MODELS_GUI_DEFAULT = "Azure GPT 4" # "GPT 4 Turbo 1106-preview"
|
| 74 |
|
| 75 |
version_mapping_cost = {
|
| 76 |
'GPT 4 32k': 'GPT_4_32K',
|
| 77 |
'GPT 4': 'GPT_4',
|
| 78 |
+
"GPT 4o 2024-05-13": 'GPT_4o_2024_05_13',
|
| 79 |
+
"GPT 4 Turbo 2024-04-09": 'GPT_4_TURBO_2024_04_09',
|
| 80 |
'GPT 4 Turbo 0125-preview': 'GPT_4_TURBO_0125',
|
| 81 |
'GPT 4 Turbo 1106-preview': 'GPT_4_TURBO_1106',
|
| 82 |
'GPT 3.5 Instruct': 'GPT_3_5_INSTRUCT',
|
| 83 |
'GPT 3.5 Turbo': 'GPT_3_5',
|
| 84 |
|
| 85 |
+
# 'Azure GPT 4 32k': 'AZURE_GPT_4_32K',
|
| 86 |
'Azure GPT 4': 'AZURE_GPT_4',
|
| 87 |
+
# 'Azure GPT 4 Turbo 0125-preview': 'AZURE_GPT_4_TURBO_0125',
|
| 88 |
+
# 'Azure GPT 4 Turbo 1106-preview': 'AZURE_GPT_4_TURBO_1106',
|
| 89 |
+
# 'Azure GPT 3.5 Instruct': 'AZURE_GPT_3_5_INSTRUCT',
|
| 90 |
+
# 'Azure GPT 3.5 Turbo': 'AZURE_GPT_3_5',
|
| 91 |
|
| 92 |
'Gemini Pro': 'GEMINI_PRO',
|
| 93 |
'PaLM 2 text-unicorn@001': 'PALM2_TU_1',
|
|
|
|
| 104 |
'LOCAL Mistral 7B Instruct v0.2': 'LOCAL_MISTRAL_7B_INSTRUCT_V02',
|
| 105 |
|
| 106 |
'LOCAL CPU Mistral 7B Instruct v0.2 GGUF': 'LOCAL_CPU_MISTRAL_7B_INSTRUCT_V02_GGUF',
|
| 107 |
+
|
| 108 |
+
'phyloforfun/mistral-7b-instruct-v2-bnb-4bit__HLT_MICH_Angiospermae_SLTPvC_v1-0_medium_OCR-C25-L25-E50-R05': 'phyloforfun/mistral-7b-instruct-v2-bnb-4bit__HLT_MICH_Angiospermae_SLTPvC_v1-0_medium_OCR-C25-L25-E50-R05',
|
| 109 |
}
|
| 110 |
|
| 111 |
@classmethod
|
| 112 |
def get_version_has_key(cls, key, has_key_openai, has_key_azure_openai, has_key_google_application_credentials, has_key_mistral):
|
| 113 |
# Define the mapping for 'has_key' values
|
| 114 |
version_has_key = {
|
| 115 |
+
"GPT 4 Turbo 2024-04-09": has_key_openai,
|
| 116 |
'GPT 4 Turbo 1106-preview': has_key_openai,
|
| 117 |
'GPT 4 Turbo 0125-preview': has_key_openai,
|
| 118 |
'GPT 4': has_key_openai,
|
| 119 |
+
"GPT 4o 2024-05-13": has_key_openai,
|
| 120 |
'GPT 4 32k': has_key_openai,
|
| 121 |
'GPT 3.5 Turbo': has_key_openai,
|
| 122 |
'GPT 3.5 Instruct': has_key_openai,
|
| 123 |
|
| 124 |
+
# 'Azure GPT 3.5 Turbo': has_key_azure_openai,
|
| 125 |
+
# 'Azure GPT 3.5 Instruct': has_key_azure_openai,
|
| 126 |
'Azure GPT 4': has_key_azure_openai,
|
| 127 |
+
# 'Azure GPT 4 Turbo 1106-preview': has_key_azure_openai,
|
| 128 |
+
# 'Azure GPT 4 Turbo 0125-preview': has_key_azure_openai,
|
| 129 |
+
# 'Azure GPT 4 32k': has_key_azure_openai,
|
| 130 |
|
| 131 |
'PaLM 2 text-bison@001': has_key_google_application_credentials,
|
| 132 |
'PaLM 2 text-bison@002': has_key_google_application_credentials,
|
|
|
|
| 143 |
'LOCAL Mistral 7B Instruct v0.2': True,
|
| 144 |
|
| 145 |
'LOCAL CPU Mistral 7B Instruct v0.2 GGUF': True,
|
| 146 |
+
|
| 147 |
+
'phyloforfun/mistral-7b-instruct-v2-bnb-4bit__HLT_MICH_Angiospermae_SLTPvC_v1-0_medium_OCR-C25-L25-E50-R05': True
|
| 148 |
}
|
| 149 |
return version_has_key.get(key)
|
| 150 |
|
| 151 |
@classmethod
|
| 152 |
def get_version_mapping_is_azure(cls, key):
|
| 153 |
version_mapping_is_azure = {
|
| 154 |
+
"GPT 4o 2024-05-13": False,
|
| 155 |
+
"GPT 4 Turbo 2024-04-09": False,
|
| 156 |
"GPT 4 Turbo 1106-preview": False,
|
| 157 |
"GPT 4 Turbo 0125-preview": False,
|
| 158 |
'GPT 4': False,
|
|
|
|
| 160 |
'GPT 3.5 Turbo': False,
|
| 161 |
'GPT 3.5 Instruct': False,
|
| 162 |
|
| 163 |
+
# 'Azure GPT 3.5 Turbo': True,
|
| 164 |
+
# 'Azure GPT 3.5 Instruct': True,
|
| 165 |
'Azure GPT 4': True,
|
| 166 |
+
# 'Azure GPT 4 Turbo 1106-preview': True,
|
| 167 |
+
# 'Azure GPT 4 Turbo 0125-preview': True,
|
| 168 |
+
# 'Azure GPT 4 32k': True,
|
| 169 |
|
| 170 |
'PaLM 2 text-bison@001': False,
|
| 171 |
'PaLM 2 text-bison@002': False,
|
|
|
|
| 182 |
'LOCAL Mistral 7B Instruct v0.2': False,
|
| 183 |
|
| 184 |
'LOCAL CPU Mistral 7B Instruct v0.2 GGUF': False,
|
| 185 |
+
|
| 186 |
+
'phyloforfun/mistral-7b-instruct-v2-bnb-4bit__HLT_MICH_Angiospermae_SLTPvC_v1-0_medium_OCR-C25-L25-E50-R05': False
|
| 187 |
}
|
| 188 |
return version_mapping_is_azure.get(key)
|
| 189 |
|
|
|
|
| 202 |
|
| 203 |
elif key == 'GPT_4_32K':
|
| 204 |
return 'gpt-4-32k'
|
| 205 |
+
|
| 206 |
+
elif key == 'GPT_4o_2024_05_13':
|
| 207 |
+
return 'gpt-4o-2024-05-13'
|
| 208 |
|
| 209 |
+
elif key == 'GPT_4_TURBO_2024_04_09':
|
| 210 |
+
return 'gpt-4-turbo-2024-04-09'
|
| 211 |
+
|
| 212 |
elif key == 'GPT_4_TURBO_1106':
|
| 213 |
return 'gpt-4-1106-preview'
|
| 214 |
|
|
|
|
| 216 |
return 'gpt-4-0125-preview'
|
| 217 |
|
| 218 |
### Azure
|
| 219 |
+
# elif key == 'AZURE_GPT_3_5':
|
| 220 |
+
# return 'gpt-35-turbo-0125'
|
| 221 |
|
| 222 |
+
# elif key == 'AZURE_GPT_3_5_INSTRUCT':
|
| 223 |
+
# return 'gpt-35-turbo-instruct'
|
| 224 |
|
| 225 |
elif key == 'AZURE_GPT_4':
|
| 226 |
return "gpt-4"
|
| 227 |
|
| 228 |
+
# elif key == 'AZURE_GPT_4_TURBO_1106':
|
| 229 |
+
# return "gpt-4-1106-preview"
|
| 230 |
|
| 231 |
+
# elif key == 'AZURE_GPT_4_TURBO_0125':
|
| 232 |
+
# return 'gpt-4-0125-preview'
|
| 233 |
|
| 234 |
+
# elif key == 'AZURE_GPT_4_32K':
|
| 235 |
+
# return "gpt-4-32k"
|
| 236 |
|
| 237 |
### Google
|
| 238 |
elif key == 'PALM2_TB_1':
|
|
|
|
| 274 |
### Mistral LOCAL CPU
|
| 275 |
elif key == 'LOCAL_CPU_MISTRAL_7B_INSTRUCT_V02_GGUF':
|
| 276 |
return 'Mistral-7B-Instruct-v0.2-GGUF'
|
| 277 |
+
|
| 278 |
+
|
| 279 |
+
|
| 280 |
+
### LOCAL custom fine-tuned
|
| 281 |
+
elif key == 'phyloforfun/mistral-7b-instruct-v2-bnb-4bit__HLT_MICH_Angiospermae_SLTPvC_v1-0_medium_OCR-C25-L25-E50-R05':
|
| 282 |
+
return 'phyloforfun/mistral-7b-instruct-v2-bnb-4bit__HLT_MICH_Angiospermae_SLTPvC_v1-0_medium_OCR-C25-L25-E50-R05'
|
| 283 |
+
|
| 284 |
+
|
| 285 |
|
| 286 |
else:
|
| 287 |
raise ValueError(f"Invalid model name {key}. See model_maps.py")
|