fix: add trust_remote_code=True to second load_dataset call in run_llama.py

improve_gainlora/src/run_llama.py

@@ -730,7 +730,9 @@ def main():
                     cache_dir=data_cache_dir,  # for debug, change dataset size, otherwise open it
                     max_num_instances_per_task=data_args.max_num_instances_per_task,
                     max_num_instances_per_eval_task=data_args.max_num_instances_per_eval_task,
-                    num_examples=data_args.num_examples)
+                    num_examples=data_args.num_examples,
+                    trust_remote_code=True,
+                )
 
                 replay_dataset_dict[task_order[idx]] = raw_datasets_gen["train"]
                 print(raw_datasets_gen)