Spaces:
Running
on
Zero
Running
on
Zero
Huakang Chen
commited on
Commit
·
be286fe
1
Parent(s):
e58ed1d
update app.py
Browse files
app.py
CHANGED
|
@@ -24,8 +24,7 @@ from xcodec2.modeling_xcodec2 import XCodec2Model
|
|
| 24 |
TEXTBOX_PLACEHOLDER = i18n("Put your text here.")
|
| 25 |
|
| 26 |
# ===== Hugging Face Model IDs =====
|
| 27 |
-
LLASA_MODEL_ID = "ASLP-lab/VoiceSculptor"
|
| 28 |
-
LLASA_SUBFOLDER = "LLaSA-Instruct-3B"
|
| 29 |
XCODEC_MODEL_ID = "HKUSTAudio/xcodec2"
|
| 30 |
PARAFORMER_REPO_ID = "funasr/Paraformer-large"
|
| 31 |
|
|
@@ -190,7 +189,7 @@ def build_app():
|
|
| 190 |
logger.info(f"✅ Loading models on device={device}")
|
| 191 |
|
| 192 |
# ===== LLaSA =====
|
| 193 |
-
tokenizer = AutoTokenizer.from_pretrained(LLASA_MODEL_ID
|
| 194 |
|
| 195 |
model = LLM(
|
| 196 |
model=LLASA_MODEL_ID,
|
|
@@ -201,8 +200,6 @@ def build_app():
|
|
| 201 |
quantization=None,
|
| 202 |
enforce_eager=False,
|
| 203 |
kv_cache_dtype="auto",
|
| 204 |
-
trust_remote_code=True,
|
| 205 |
-
hf_model_subfolder=LLASA_SUBFOLDER,
|
| 206 |
)
|
| 207 |
|
| 208 |
# ===== XCodec2 =====
|
|
|
|
| 24 |
TEXTBOX_PLACEHOLDER = i18n("Put your text here.")
|
| 25 |
|
| 26 |
# ===== Hugging Face Model IDs =====
|
| 27 |
+
LLASA_MODEL_ID = "ASLP-lab/VoiceSculptor-VoiceDesign"
|
|
|
|
| 28 |
XCODEC_MODEL_ID = "HKUSTAudio/xcodec2"
|
| 29 |
PARAFORMER_REPO_ID = "funasr/Paraformer-large"
|
| 30 |
|
|
|
|
| 189 |
logger.info(f"✅ Loading models on device={device}")
|
| 190 |
|
| 191 |
# ===== LLaSA =====
|
| 192 |
+
tokenizer = AutoTokenizer.from_pretrained(LLASA_MODEL_ID)
|
| 193 |
|
| 194 |
model = LLM(
|
| 195 |
model=LLASA_MODEL_ID,
|
|
|
|
| 200 |
quantization=None,
|
| 201 |
enforce_eager=False,
|
| 202 |
kv_cache_dtype="auto",
|
|
|
|
|
|
|
| 203 |
)
|
| 204 |
|
| 205 |
# ===== XCodec2 =====
|