Update README.md
Browse files
README.md
CHANGED
@@ -49,7 +49,7 @@ model.to('cuda')
|
|
49 |
|
50 |
from xcodec2.modeling_xcodec2 import XCodec2Model
|
51 |
|
52 |
-
model_path = "
|
53 |
|
54 |
Codec_model = XCodec2Model.from_pretrained(model_path)
|
55 |
Codec_model.eval().cuda()
|
@@ -129,7 +129,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
129 |
import torch
|
130 |
import soundfile as sf
|
131 |
|
132 |
-
llasa_1b ='
|
133 |
|
134 |
tokenizer = AutoTokenizer.from_pretrained(llasa_1b)
|
135 |
model = AutoModelForCausalLM.from_pretrained(llasa_1b)
|
@@ -138,7 +138,7 @@ model.to('cuda')
|
|
138 |
|
139 |
from xcodec2.modeling_xcodec2 import XCodec2Model
|
140 |
|
141 |
-
model_path = "
|
142 |
|
143 |
Codec_model = XCodec2Model.from_pretrained(model_path)
|
144 |
Codec_model.eval().cuda()
|
|
|
49 |
|
50 |
from xcodec2.modeling_xcodec2 import XCodec2Model
|
51 |
|
52 |
+
model_path = "HKUSTAudio/xcodec2"
|
53 |
|
54 |
Codec_model = XCodec2Model.from_pretrained(model_path)
|
55 |
Codec_model.eval().cuda()
|
|
|
129 |
import torch
|
130 |
import soundfile as sf
|
131 |
|
132 |
+
llasa_1b ='HKUSTAudio/Llasa-1b'
|
133 |
|
134 |
tokenizer = AutoTokenizer.from_pretrained(llasa_1b)
|
135 |
model = AutoModelForCausalLM.from_pretrained(llasa_1b)
|
|
|
138 |
|
139 |
from xcodec2.modeling_xcodec2 import XCodec2Model
|
140 |
|
141 |
+
model_path = "HKUSTAudio/xcodec2"
|
142 |
|
143 |
Codec_model = XCodec2Model.from_pretrained(model_path)
|
144 |
Codec_model.eval().cuda()
|