Elgato_dark/chronos-forecasting
1
0
Fork 0
mirror of https://github.com/amazon-science/chronos-forecasting synced 2026-05-24 01:58:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add: If dict is given

Browse source
This commit is contained in:
Ryuichi Ichinose 2025-12-13 16:35:24 +09:00
parent 7f2f5a1d18
commit 6248da86ec
No known key found for this signature in database
GPG key ID: D5D8D056D6A7CA53
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
src/chronos/chronos2/pipeline.py
View file

@ -210,6 +210,7 @@ class Chronos2Pipeline(BaseChronosPipeline):
revision=lora_revision,
)
elif isinstance(lora_config, dict):
lora_config.setdefault("revision", lora_revision)
lora_config = LoraConfig(**lora_config)
else:
assert isinstance(lora_config, LoraConfig), (