Bram Vanroy
add data collection script
828458d
raw
history blame
2.82 kB
{
"bloom-7b1": {
"model_name": "pretrained=bigscience/bloom-7b1",
"args": "pretrained=bigscience/bloom-7b1"
},
"gpt-neo-1.3b-dutch": {
"model_name": "yhavinga/gpt-neo-1.3B-dutch",
"args": "use_accelerate=True,device_map_option=auto,dtype=bfloat16,load_in_8bit=True"
},
"gpt-neo-125m-dutch": {
"model_name": "yhavinga/gpt-neo-125M-dutch",
"args": "use_accelerate=True,device_map_option=auto,dtype=bfloat16,load_in_8bit=True"
},
"gpt2-large-dutch": {
"model_name": "yhavinga/gpt2-large-dutch",
"args": "use_accelerate=True,device_map_option=auto,dtype=bfloat16,load_in_8bit=True"
},
"gpt2-medium-dutch": {
"model_name": "yhavinga/gpt2-medium-dutch",
"args": "use_accelerate=True,device_map_option=auto,dtype=bfloat16,load_in_8bit=True"
},
"llama-2-13b-chat-dutch": {
"model_name": "BramVanroy/Llama-2-13b-chat-dutch",
"args": "use_accelerate=True,device_map_option=auto,dtype=bfloat16,load_in_8bit=True"
},
"llama-2-13b-chat-hf": {
"model_name": "meta-llama/Llama-2-13b-chat-hf",
"args": "use_accelerate=True,device_map_option=auto,dtype=bfloat16,load_in_8bit=True"
},
"llama-2-13b-hf": {
"model_name": "meta-llama/Llama-2-13b-hf",
"args": "use_accelerate=True,device_map_option=auto,dtype=bfloat16,load_in_8bit=True"
},
"llama-2-7b-chat-hf": {
"model_name": "meta-llama/Llama-2-7b-chat-hf",
"args": "use_accelerate=True,device_map_option=auto,dtype=bfloat16,load_in_8bit=True"
},
"llama-2-7b-hf": {
"model_name": "meta-llama/Llama-2-7b-hf",
"args": "use_accelerate=True,device_map_option=auto,dtype=bfloat16,load_in_8bit=True"
},
"llama-7b": {
"model_name": "pretrained=/sensei-fs/users/daclai/uoChatGPT/llama-7B",
"args": "pretrained=/sensei-fs/users/daclai/uoChatGPT/llama-7B"
},
"llama2-13b-ft-mc4_nl_cleaned_tiny": {
"model_name": "BramVanroy/llama2-13b-ft-mc4_nl_cleaned_tiny",
"args": "use_accelerate=True,device_map_option=auto,dtype=bfloat16,load_in_8bit=True"
},
"mistral-7b-v0.1": {
"model_name": "mistralai/Mistral-7B-v0.1",
"args": "use_accelerate=True,device_map_option=auto,dtype=bfloat16,load_in_8bit=True"
},
"neural-chat-7b-v3-1": {
"model_name": "Intel/neural-chat-7b-v3-1",
"args": "use_accelerate=True,device_map_option=auto,dtype=bfloat16,load_in_8bit=True"
},
"orca-2-13b": {
"model_name": "microsoft/Orca-2-13b",
"args": "use_accelerate=True,device_map_option=auto,dtype=bfloat16,load_in_8bit=True"
},
"orca-2-7b": {
"model_name": "microsoft/Orca-2-7b",
"args": "use_accelerate=True,device_map_option=auto,dtype=bfloat16,load_in_8bit=True"
},
"zephyr-7b-beta": {
"model_name": "HuggingFaceH4/zephyr-7b-beta",
"args": "use_accelerate=True,device_map_option=auto,dtype=bfloat16,load_in_8bit=True"
}
}