Update configuration_moss_audio_tokenizer.py

#1
by fablevi - opened
Files changed (1) hide show
  1. configuration_moss_audio_tokenizer.py +11 -11
configuration_moss_audio_tokenizer.py CHANGED
@@ -86,17 +86,17 @@ class MossAudioTokenizerConfig(PreTrainedConfig):
86
  # Backward-compatible alias used by some checkpoints.
87
  attribute_map = {"sample_rate": "sampling_rate"}
88
 
89
- sampling_rate: int
90
- downsample_rate: int
91
- causal_transformer_context_duration: float
92
- encoder_kwargs: list[dict[str, Any]]
93
- decoder_kwargs: list[dict[str, Any]]
94
- number_channels: int
95
- enable_channel_interleave: bool
96
- attention_implementation: str
97
- compute_dtype: str
98
- quantizer_type: str
99
- quantizer_kwargs: dict[str, Any]
100
 
101
  def __init__(
102
  self,
 
86
  # Backward-compatible alias used by some checkpoints.
87
  attribute_map = {"sample_rate": "sampling_rate"}
88
 
89
+ sampling_rate: int = 48000
90
+ downsample_rate: int = 3840
91
+ causal_transformer_context_duration: float = 10.0
92
+ encoder_kwargs: list = None
93
+ decoder_kwargs: list = None
94
+ number_channels: int = 2
95
+ enable_channel_interleave: bool = True
96
+ attention_implementation: str = "sdpa"
97
+ compute_dtype: str = "fp32"
98
+ quantizer_type: str = "rlfq"
99
+ quantizer_kwargs: dict = None
100
 
101
  def __init__(
102
  self,