Trust remote code

allenai · Oct 30, 2024 · d45b34f · d45b34f
1 parent cda0ad7
commit d45b34f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pdelfin/train/loaddataset.py b/pdelfin/train/loaddataset.py
@@ -15,7 +15,7 @@
 def main():
     train_config = make_cli(TrainConfig)  # pyright: ignore
 
-    processor = AutoProcessor.from_pretrained(train_config.model.name_or_path)
+    processor = AutoProcessor.from_pretrained(train_config.model.name_or_path, trust_remote_code=True)
     train_dataset, valid_dataset = make_dataset(train_config, processor)    
 
     print("Training dataset........")