always trust remote code

ZYC-ModelCloud · Apr 25, 2023 · 419160b · 419160b
1 parent cfcba7d
commit 419160b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/auto_gptq/modeling/_base.py b/auto_gptq/modeling/_base.py
@@ -357,7 +357,7 @@ def skip(*args, **kwargs):
 
         transformers.modeling_utils._init_weights = False
         torch.set_default_dtype(torch.half)
-        model = AutoModelForCausalLM.from_config(config)
+        model = AutoModelForCausalLM.from_config(config, trust_remote_code=True)
         torch.set_default_dtype(torch.float)
         model = model.eval()
         layers = find_layers(model)