mirror of
https://github.com/hanzoai/mugen.git
synced 2026-10-30 18:27:38 +00:00
20 lines
655 B
Python
20 lines
655 B
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
|
|
from . import builders, musicgen
|
|
|
|
|
|
class AudioGenSolver(musicgen.MusicGenSolver):
|
|
"""Solver for AudioGen re-implementation training task.
|
|
|
|
Note that this implementation does not strictly follows
|
|
the method proposed in https://arxiv.org/abs/2209.15352
|
|
but is derived from MusicGen's training pipeline.
|
|
|
|
More information can be found in the AudioGen model card.
|
|
"""
|
|
DATASET_TYPE: builders.DatasetType = builders.DatasetType.SOUND
|