diff --git a/examples/huggingface/bert_influence.py b/examples/huggingface/bert_influence.py index a2fb238..ea3d8ff 100644 --- a/examples/huggingface/bert_influence.py +++ b/examples/huggingface/bert_influence.py @@ -7,7 +7,7 @@ from transformers import default_data_collator, Trainer, TrainingArguments import logix -from logix.huggingface import patch_trainer, LogIXArgument +from logix.huggingface import patch_trainer, LogIXArguments from bert_utils import construct_model, get_datasets, set_seed @@ -28,7 +28,7 @@ def main(): model.eval() train_dataset = get_datasets(args.data_name)[-1] - logix_args = LogIXArgument( + logix_args = LogIXArguments( project=args.project, config=args.config_path, lora=True, diff --git a/examples/huggingface/bert_log.py b/examples/huggingface/bert_log.py index 2dc6238..70671ee 100644 --- a/examples/huggingface/bert_log.py +++ b/examples/huggingface/bert_log.py @@ -6,7 +6,7 @@ from transformers import default_data_collator, Trainer, TrainingArguments import logix -from logix.huggingface import patch_trainer, LogIXArgument +from logix.huggingface import patch_trainer, LogIXArguments from bert_utils import construct_model, get_datasets, set_seed @@ -27,7 +27,7 @@ def main(): model.eval() train_dataset = get_datasets(args.data_name)[1] - logix_args = LogIXArgument( + logix_args = LogIXArguments( project=args.project, config=args.config_path, lora=True, diff --git a/examples/huggingface/gpt_influence.py b/examples/huggingface/gpt_influence.py index db76984..26ffa09 100644 --- a/examples/huggingface/gpt_influence.py +++ b/examples/huggingface/gpt_influence.py @@ -7,7 +7,7 @@ from transformers import default_data_collator, Trainer, TrainingArguments import logix -from logix.huggingface import patch_trainer, LogIXArgument +from logix.huggingface import patch_trainer, LogIXArguments from gpt_utils import construct_model, get_datasets, set_seed @@ -26,7 +26,7 @@ def main(): model.eval() train_dataset = get_datasets()[-1] - logix_args = LogIXArgument( + logix_args = LogIXArguments( project=args.project, config=args.config_path, lora=True, diff --git a/examples/huggingface/gpt_log.py b/examples/huggingface/gpt_log.py index 828e8ac..db24109 100644 --- a/examples/huggingface/gpt_log.py +++ b/examples/huggingface/gpt_log.py @@ -6,7 +6,7 @@ from transformers import default_data_collator, Trainer, TrainingArguments import logix -from logix.huggingface import patch_trainer, LogIXArgument +from logix.huggingface import patch_trainer, LogIXArguments from gpt_utils import construct_model, get_datasets, set_seed @@ -25,7 +25,7 @@ def main(): model.eval() train_dataset = get_datasets()[0] - logix_args = LogIXArgument( + logix_args = LogIXArguments( project=args.project, config=args.config_path, lora=True,