support IPEX

Author: DDXDB

cosyvoice/cli/cosyvoice.py

@@ -18,6 +18,11 @@
 from hyperpyyaml import load_hyperpyyaml
 from modelscope import snapshot_download
 import torch
+try:
+    import intel_extension_for_pytorch as ipex
+except Exception:
+    pass
+
 from cosyvoice.cli.frontend import CosyVoiceFrontEnd
 from cosyvoice.cli.model import CosyVoiceModel, CosyVoice2Model
 from cosyvoice.utils.file_utils import logging

cosyvoice/cli/frontend.py

@@ -16,6 +16,11 @@
 import json
 import onnxruntime
 import torch
+try:
+    import intel_extension_for_pytorch as ipex
+except Exception:
+    pass
+
 import numpy as np
 import whisper
 from typing import Callable

cosyvoice/cli/model.py

@@ -14,6 +14,11 @@
 import os
 from typing import Generator
 import torch
+try:
+    import intel_extension_for_pytorch as ipex
+except Exception:
+    pass
+
 import numpy as np
 import threading
 import time

requirements.txt

@@ -35,4 +35,4 @@ uvicorn==0.30.0
 wget==3.2
 fastapi==0.115.6
 fastapi-cli==0.0.4
-WeTextProcessing==1.0.3
+WeTextProcessing==1.0.3