lyuxiang.lx 5 mēneši atpakaļ
vecāks
revīzija
21a5efd8ae
2 mainītis faili ar 2 papildinājumiem un 2 dzēšanām
  1. 1 1
      README.md
  2. 1 1
      cosyvoice/llm/llm.py

+ 1 - 1
README.md

@@ -72,7 +72,7 @@
     # pynini is required by WeTextProcessing, use conda to install it as it can be executed on all platforms.
     conda install -y -c conda-forge pynini==2.1.5
     pip install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/ --trusted-host=mirrors.aliyun.com
-    
+
     # If you encounter sox compatibility issues
     # ubuntu
     sudo apt-get install sox libsox-dev

+ 1 - 1
cosyvoice/llm/llm.py

@@ -381,7 +381,7 @@ class Qwen2LM(TransformerLM):
             self,
             batch: dict,
             device: torch.device,
-        ) -> Dict[str, Optional[torch.Tensor]]:
+    ) -> Dict[str, Optional[torch.Tensor]]:
         text_token = batch['text_token'].to(device)
         text_token_len = batch['text_token_len'].to(device)
         speech_token = batch['speech_token'].to(device)