This commit is contained in:
Alex Barron
2024-10-09 09:51:16 -07:00
parent 4d2ee67402
commit 6a37b5106a
3 changed files with 2 additions and 5 deletions

1
musicgen/encodec.py Symbolic link
View File

@@ -0,0 +1 @@
../encodec/encodec.py

View File

@@ -1,7 +1,6 @@
# Copyright © 2024 Apple Inc.
import json
import sys
from functools import partial
from pathlib import Path
from types import SimpleNamespace
@@ -11,10 +10,6 @@ import mlx.core as mx
import mlx.nn as nn
from tqdm import tqdm
cur_path = Path(__file__).parents[1].resolve()
sys.path.append(str(cur_path / "encodec"))
sys.path.append(str(cur_path / "t5"))
from encodec import EncodecModel
from t5 import T5

1
musicgen/t5.py Symbolic link
View File

@@ -0,0 +1 @@
../t5/t5.py