Huakang Chen commited on
Commit
be286fe
·
1 Parent(s): e58ed1d

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
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, subfolder=LLASA_SUBFOLDER, trust_remote_code=True)
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 =====