660acc7
1
2
3
4
5
6
7
8
9
10
11
12
13
14
# Author: Bingxin Ke # Last modified: 2024-05-17 from .marigold_trainer import MarigoldTrainer trainer_cls_name_dict = { "MarigoldTrainer": MarigoldTrainer, } def get_trainer_cls(trainer_name): return trainer_cls_name_dict[trainer_name]
