tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Fix use_mlflow to be bool instead of str (#1344)

Browse Source
This commit is contained in:
Chirag Jain
2024-02-28 23:28:29 +05:30
committed by GitHub
parent 6d4bbb877f
commit 3a5a2d2f34
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/utils/config/models/input/v0_4_1/__init__.py
View File

@@ -322,7 +322,7 @@ class ModelOutputConfig(BaseModel):
class MLFlowConfig(BaseModel):
"""mlflow configuration subset"""
use_mlflow: Optional[str] = None
use_mlflow: Optional[bool] = None
mlflow_tracking_uri: Optional[str] = None
mlflow_experiment_name: Optional[str] = None
hf_mlflow_log_artifacts: Optional[bool] = None