Skip to content

Commit d0ff30e

Browse files
committed
Allow loading only encoder or decoder
1 parent e71902b commit d0ff30e

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/models.js

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7831,6 +7831,11 @@ const CUSTOM_MAPPING = [
78317831
['JinaCLIPTextModel', JinaCLIPTextModel, MODEL_TYPES.EncoderOnly],
78327832
['ClapTextModelWithProjection', ClapTextModelWithProjection, MODEL_TYPES.EncoderOnly],
78337833
['ClapAudioModelWithProjection', ClapAudioModelWithProjection, MODEL_TYPES.EncoderOnly],
7834+
7835+
['DacEncoderModel', DacEncoderModel, MODEL_TYPES.EncoderOnly],
7836+
['DacDecoderModel', DacDecoderModel, MODEL_TYPES.EncoderOnly],
7837+
['MimiEncoderModel', MimiEncoderModel, MODEL_TYPES.EncoderOnly],
7838+
['MimiDecoderModel', MimiDecoderModel, MODEL_TYPES.EncoderOnly],
78347839
]
78357840
for (const [name, model, type] of CUSTOM_MAPPING) {
78367841
MODEL_TYPE_MAPPING.set(name, type);

0 commit comments

Comments
 (0)