Update README.md
Browse files
README.md
CHANGED
|
@@ -62,7 +62,7 @@ model.to('cuda')
|
|
| 62 |
|
| 63 |
from xcodec2.modeling_xcodec2 import XCodec2Model
|
| 64 |
|
| 65 |
-
model_path = "
|
| 66 |
|
| 67 |
Codec_model = XCodec2Model.from_pretrained(model_path)
|
| 68 |
Codec_model.eval().cuda()
|
|
@@ -144,7 +144,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
| 144 |
import torch
|
| 145 |
import soundfile as sf
|
| 146 |
|
| 147 |
-
llasa_8b ='
|
| 148 |
|
| 149 |
tokenizer = AutoTokenizer.from_pretrained(llasa_8b)
|
| 150 |
model = AutoModelForCausalLM.from_pretrained(llasa_8b)
|
|
@@ -153,7 +153,7 @@ model.to('cuda')
|
|
| 153 |
|
| 154 |
from xcodec2.modeling_xcodec2 import XCodec2Model
|
| 155 |
|
| 156 |
-
model_path = "
|
| 157 |
|
| 158 |
Codec_model = XCodec2Model.from_pretrained(model_path)
|
| 159 |
Codec_model.eval().cuda()
|
|
|
|
| 62 |
|
| 63 |
from xcodec2.modeling_xcodec2 import XCodec2Model
|
| 64 |
|
| 65 |
+
model_path = "HKUSTAudio/xcodec2"
|
| 66 |
|
| 67 |
Codec_model = XCodec2Model.from_pretrained(model_path)
|
| 68 |
Codec_model.eval().cuda()
|
|
|
|
| 144 |
import torch
|
| 145 |
import soundfile as sf
|
| 146 |
|
| 147 |
+
llasa_8b ='HKUSTAudio/Llasa-8B'
|
| 148 |
|
| 149 |
tokenizer = AutoTokenizer.from_pretrained(llasa_8b)
|
| 150 |
model = AutoModelForCausalLM.from_pretrained(llasa_8b)
|
|
|
|
| 153 |
|
| 154 |
from xcodec2.modeling_xcodec2 import XCodec2Model
|
| 155 |
|
| 156 |
+
model_path = "HKUSTAudio/xcodec2"
|
| 157 |
|
| 158 |
Codec_model = XCodec2Model.from_pretrained(model_path)
|
| 159 |
Codec_model.eval().cuda()
|