diff --git a/GPT_SoVITS/TTS_infer_pack/TTS.py b/GPT_SoVITS/TTS_infer_pack/TTS.py
index b38e9dc..4befc0c 100644
--- a/GPT_SoVITS/TTS_infer_pack/TTS.py
+++ b/GPT_SoVITS/TTS_infer_pack/TTS.py
@@ -749,7 +749,7 @@ class TTS:
                 if no_prompt_text :
                     prompt = None
                 else:
-                    prompt = self.prompt_cache["prompt_semantic"].expand(all_phoneme_ids.shape[0], -1).to(self.configs.device)
+                    prompt = self.prompt_cache["prompt_semantic"].expand(len(all_phoneme_ids), -1).to(self.configs.device)
 
 
                 pred_semantic_list, idx_list = self.t2s_model.model.infer_panel(
diff --git a/colab_webui.ipynb b/colab_webui.ipynb
index 0d50fe1..838f826 100644
--- a/colab_webui.ipynb
+++ b/colab_webui.ipynb
@@ -67,7 +67,7 @@
         "!git clone https://www.modelscope.cn/damo/punc_ct-transformer_zh-cn-common-vocab272727-pytorch.git\n",
         "# @title UVR5 pretrains 安装uvr5模型\n",
         "%cd /content/GPT-SoVITS/tools/uvr5\n",
-        "%rm -r uvr5_weights\n"，
+        "%rm -r uvr5_weights\n",
         "!git clone https://huggingface.co/Delik/uvr5_weights\n",
         "!git config core.sparseCheckout true\n",
         "!mv /content/GPT-SoVITS/GPT_SoVITS/pretrained_models/GPT-SoVITS/* /content/GPT-SoVITS/GPT_SoVITS/pretrained_models/"