Chronos-2: Handle missing 'peft' and lora_config specified (#421)

*Issue #, if available:*

*Description of changes:*


By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice.
This commit is contained in:
Abdul Fatir 2025-12-04 10:43:56 +01:00 committed by GitHub
parent 564218776a
commit 89031842fa
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -178,6 +178,7 @@ class Chronos2Pipeline(BaseChronosPipeline):
"`peft` is required for `finetune_mode='lora'`. Please install it with `pip install peft`. Falling back to `finetune_mode='full'`."
)
finetune_mode = "full"
lora_config = None
from chronos.chronos2.trainer import Chronos2Trainer, EvaluateAndSaveFinalStepCallback