temp commit

This commit is contained in:
Botao Chen 2024-12-02 17:35:40 -08:00
parent 79c525be94
commit 745163b6b0

View file

@ -186,7 +186,6 @@ class PostTraining(Protocol):
validation_dataset_id: str, validation_dataset_id: str,
algorithm: FinetuningAlgorithm, algorithm: FinetuningAlgorithm,
algorithm_config: LoraFinetuningConfig, algorithm_config: LoraFinetuningConfig,
# optimizer_config: OptimizerConfig,
training_config: TrainingConfig, training_config: TrainingConfig,
hyperparam_search_config: Dict[str, Any], hyperparam_search_config: Dict[str, Any],
logger_config: Dict[str, Any], logger_config: Dict[str, Any],