Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-09 15:02:01 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix deprecation warning

Browse source
This commit is contained in:
Pierre Chapuis 2024-08-28 10:39:41 +02:00
parent e643006887
commit 6f8e244155
No known key found for this signature in database
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/refiners/training_utils/config.py
View file

@ -1,5 +1,5 @@
from enum import Enum from enum import Enum
from logging import warn from logging import warning
from pathlib import Path from pathlib import Path
from typing import Annotated, Any, Callable, Iterable, Literal, Type, TypeVar from typing import Annotated, Any, Callable, Iterable, Literal, Type, TypeVar
@ -132,7 +132,7 @@ class OptimizerConfig(BaseModel):
) )
case Optimizers.Prodigy: case Optimizers.Prodigy:
if self.learning_rate != 1.0: if self.learning_rate != 1.0:
warn("Prodigy learning rate is not 1.0, this might cause instability.") warning("Prodigy learning rate is not 1.0, this might cause instability.")
return Prodigy( return Prodigy(
lr=self.learning_rate, lr=self.learning_rate,
params=params, params=params,