Skip to content

Commit defd290

Browse files
small fixes for args and paper link
1 parent 0b0958c commit defd290

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

examples/orthogonal_subspace_learning/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -292,7 +292,7 @@ If you use OSF in your research, please cite:
292292

293293
- [OSF Documentation](../../docs/source/package_reference/osf.md)
294294
- [PEFT Documentation](https://huggingface.co/docs/peft)
295-
- [Original Paper](https://arxiv.org/abs/2504.07097)
295+
- [Original Paper](https://huggingface.co/papers/2504.07097)
296296

297297
## License
298298

examples/orthogonal_subspace_learning/osf_continual_learning.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -194,6 +194,9 @@ def evaluate_model(model, eval_dataset, data_collator, tokenizer, task_name, tas
194194
model=model,
195195
data_collator=data_collator,
196196
eval_dataset=eval_dataset,
197+
args=TrainingArguments(
198+
label_names=["labels"],
199+
),
197200
)
198201
results = trainer.evaluate()
199202
loss = results["eval_loss"]
@@ -234,7 +237,7 @@ def train_with_osf(
234237
tokenizer.pad_token = tokenizer.eos_token
235238

236239
base_model = AutoModelForCausalLM.from_pretrained(
237-
model_name, torch_dtype=torch.bfloat16, device_map="auto", trust_remote_code=True
240+
model_name, torch_dtype=torch.bfloat16, device_map="auto"
238241
)
239242

240243
# Load all datasets with task-specific sizes

0 commit comments

Comments
 (0)