[Bugfix] Deprecate registration of custom configs to huggingface (#9083)

2024-10-05 06:56:40 -07:00
parent 15986f598c
commit cfadb9c687
3 changed files with 2 additions and 16 deletions
--- a/tests/models/decoder_only/vision_language/test_internvl.py
+++ b/tests/models/decoder_only/vision_language/test_internvl.py
@@ -97,7 +97,8 @@ def run_test(
            self.tokenizer = hf_runner.tokenizer
            self.dtype = hf_runner.model.dtype

-            self.config = AutoConfig.from_pretrained(hf_runner.model_name)
+            self.config = AutoConfig.from_pretrained(hf_runner.model_name,
+                                                     trust_remote_code=True)
            self.vision_config = self.config.vision_config
            self.use_thumbnail = self.config.use_thumbnail
            self.min_num = self.config.min_dynamic_patch