From 6b9a5e4b68de6428678e10cf9b8d731c48aad5a9 Mon Sep 17 00:00:00 2001 From: Travis Addair Date: Fri, 4 Aug 2023 11:59:06 -0700 Subject: [PATCH] Remove falcon from presets as it requires running untrusted code (#3495) --- ludwig/schema/llms/base_model.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/ludwig/schema/llms/base_model.py b/ludwig/schema/llms/base_model.py index 6d17d0f26d4..9df99de4857 100644 --- a/ludwig/schema/llms/base_model.py +++ b/ludwig/schema/llms/base_model.py @@ -19,10 +19,6 @@ "llama-2-7b-chat": "meta-llama/Llama-2-7b-chat-hf", "llama-2-13b-chat": "meta-llama/Llama-2-13b-chat-hf", "llama-2-70b-chat": "meta-llama/Llama-2-70b-chat-hf", - "falcon-7b": "tiiuae/falcon-7b", - "falcon-40b": "tiiuae/falcon-40b", - "falcon-7b-instruct": "tiiuae/falcon-7b-instruct", - "falcon-40b-instruct": "tiiuae/falcon-40b-instruct", "vicuna-7b": "lmsys/vicuna-7b-v1.3", "vicuna-13b": "lmsys/vicuna-13b-v1.3", "bloomz-3b": "bigscience/bloomz-3b",