text: from keras_cv_attention_models.model_surgery.model_surgery import (
      SAMModel,
      DropConnect,
      add_l2_regularizer_2_model,
      convert_to_mixed_float16,
      convert_mixed_float16_to_float32,
      convert_to_fused_conv_bn_model,
      get_actual_survival_probabilities,
      get_actual_drop_connect_rates,
      replace_ReLU,
      replace_add_with_drop_connect,
      replace_add_with_stochastic_depth,
      replace_stochastic_depth_with_add,
  )
  