content: # from CoTrain.modules.cotrain_dino_module_v2 import CoTrainTransformerSS
  from CoTrain.modules.cotrain_module import CoTrainTransformerSS
  # from CoTrain.modules.cotrain_dino_module_v3 import CoTrainTransformerSS
  from CoTrain.modules.clip_module import CLIP