mirror of
https://github.com/amazon-science/chronos-forecasting
synced 2026-05-24 10:08:33 +00:00
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:
parent
564218776a
commit
89031842fa
1 changed files with 1 additions and 0 deletions
|
|
@ -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
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue