Commit
•
28d2e51
1
Parent(s):
26b4608
Create models/__init__.py
Browse files
audiocraft/models/__init__.py
ADDED
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
2 |
+
# All rights reserved.
|
3 |
+
#
|
4 |
+
# This source code is licensed under the license found in the
|
5 |
+
# LICENSE file in the root directory of this source tree.
|
6 |
+
"""
|
7 |
+
Models for EnCodec, AudioGen, MusicGen, as well as the generic LMModel.
|
8 |
+
"""
|
9 |
+
# flake8: noqa
|
10 |
+
from . import builders, loaders
|
11 |
+
from .encodec import (
|
12 |
+
CompressionModel, EncodecModel, DAC,
|
13 |
+
HFEncodecModel, HFEncodecCompressionModel)
|
14 |
+
from .audiogen import AudioGen
|
15 |
+
from .lm import LMModel
|
16 |
+
from .lm_magnet import MagnetLMModel
|
17 |
+
from .multibanddiffusion import MultiBandDiffusion
|
18 |
+
from .musicgen import MusicGen
|
19 |
+
from .magnet import MAGNeT
|
20 |
+
from .unet import DiffusionUnet
|