Add french, italian, portuguese, spanish, german (#155)
The french 6 layers variant wasn't included because it's not good enough yet. Use the `french_24l` variant for now.
This commit is contained in:
@@ -22,6 +22,8 @@ class MimiModel(nn.Module):
|
||||
encoder_frame_rate: float,
|
||||
sample_rate: int,
|
||||
channels: int,
|
||||
inner_dim: int | None,
|
||||
outer_dim: int | None,
|
||||
encoder_transformer: ProjectedTransformer,
|
||||
decoder_transformer: ProjectedTransformer,
|
||||
):
|
||||
@@ -50,8 +52,14 @@ class MimiModel(nn.Module):
|
||||
assert downsample_stride == int(downsample_stride), (
|
||||
f"Only integer strides are supported, got {downsample_stride}"
|
||||
)
|
||||
self.downsample = ConvDownsample1d(int(downsample_stride), dimension=dimension)
|
||||
self.upsample = ConvTrUpsample1d(int(downsample_stride), dimension=dimension)
|
||||
|
||||
# v2 of models
|
||||
self.downsample = ConvDownsample1d(
|
||||
int(downsample_stride), dimension=dimension, out_dimension=inner_dim
|
||||
)
|
||||
self.upsample = ConvTrUpsample1d(
|
||||
int(downsample_stride), dimension=dimension, in_dimension=outer_dim
|
||||
)
|
||||
|
||||
@property
|
||||
def frame_size(self) -> int:
|
||||
|
||||
Reference in New Issue
Block a user