config.json

{
  "architectures": [
    "BearingsModel"
  ],
  "auto_map": {
    "AutoConfig": "configuration_bearings.BearingsConfig",
    "AutoFeatureExtractor": "feature_extraction_bearings.BearingsFeatureExtractor",
    "AutoModel": "modeling_bearings.BearingsModel"
  },
  "dtype": "float32",
  "encoder_depth": 6,
  "encoder_embedding_dim": 384,
  "f_max": 14000.0,
  "f_min": 50.0,
  "fshape": 16,
  "hop_length": 320,
  "in_channels": 7,
  "layer_norm_eps": 1e-05,
  "mlp_ratio": 4.0,
  "model_type": "bearings",
  "n_fft": 1024,
  "num_heads": 12,
  "num_mel_bins": 128,
  "sampling_rate": 32000,
  "target_length": 200,
  "transformers_version": "4.57.6",
  "tshape": 8,
  "win_length": 1024
}