Update README.md
Browse files
README.md
CHANGED
@@ -52,7 +52,7 @@ model.to('cuda')
|
|
52 |
|
53 |
from xcodec2.modeling_xcodec2 import XCodec2Model
|
54 |
|
55 |
-
model_path = "
|
56 |
|
57 |
Codec_model = XCodec2Model.from_pretrained(model_path)
|
58 |
Codec_model.eval().cuda()
|
@@ -132,7 +132,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
132 |
import torch
|
133 |
import soundfile as sf
|
134 |
|
135 |
-
llasa_3b ='
|
136 |
|
137 |
tokenizer = AutoTokenizer.from_pretrained(llasa_3b)
|
138 |
model = AutoModelForCausalLM.from_pretrained(llasa_3b)
|
@@ -141,7 +141,7 @@ model.to('cuda')
|
|
141 |
|
142 |
from xcodec2.modeling_xcodec2 import XCodec2Model
|
143 |
|
144 |
-
model_path = "
|
145 |
|
146 |
Codec_model = XCodec2Model.from_pretrained(model_path)
|
147 |
Codec_model.eval().cuda()
|
|
|
52 |
|
53 |
from xcodec2.modeling_xcodec2 import XCodec2Model
|
54 |
|
55 |
+
model_path = "HKUSTAudio/xcodec2"
|
56 |
|
57 |
Codec_model = XCodec2Model.from_pretrained(model_path)
|
58 |
Codec_model.eval().cuda()
|
|
|
132 |
import torch
|
133 |
import soundfile as sf
|
134 |
|
135 |
+
llasa_3b ='HKUSTAudio/Llasa-3B'
|
136 |
|
137 |
tokenizer = AutoTokenizer.from_pretrained(llasa_3b)
|
138 |
model = AutoModelForCausalLM.from_pretrained(llasa_3b)
|
|
|
141 |
|
142 |
from xcodec2.modeling_xcodec2 import XCodec2Model
|
143 |
|
144 |
+
model_path = "HKUSTAudio/xcodec2"
|
145 |
|
146 |
Codec_model = XCodec2Model.from_pretrained(model_path)
|
147 |
Codec_model.eval().cuda()
|