From 62d59a516f6b5610f4e10d45ebe1e7a4202024df Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Fri, 8 Dec 2023 06:29:26 -0800 Subject: [PATCH] Add trust_remote_code to all HF loaders --- modules/loaders.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/modules/loaders.py b/modules/loaders.py index babbe440..d1f8343f 100644 --- a/modules/loaders.py +++ b/modules/loaders.py @@ -59,6 +59,7 @@ loaders_and_params = OrderedDict({ 'cpu', 'numa', 'cfg_cache', + 'trust_remote_code', 'no_use_fast', 'logits_all', 'llamacpp_HF_info', @@ -70,6 +71,7 @@ loaders_and_params = OrderedDict({ 'rope_freq_base', 'compress_pos_emb', 'cfg_cache', + 'trust_remote_code', 'no_use_fast', ], 'ExLlamav2_HF': [ @@ -80,6 +82,7 @@ loaders_and_params = OrderedDict({ 'cache_8bit', 'alpha_value', 'compress_pos_emb', + 'trust_remote_code', 'no_use_fast', ], 'AutoGPTQ': [ @@ -114,6 +117,7 @@ loaders_and_params = OrderedDict({ 'groupsize', 'model_type', 'pre_layer', + 'trust_remote_code', 'no_use_fast', 'gptq_for_llama_info', ],