Sfoglia il codice sorgente

Intel arc XPU support (#1033)

* XPU support + install docs

* update docs with conda necessary lib
Nacholmo 10 mesi fa
parent
commit
6945784348
6 ha cambiato i file con 67 aggiunte e 0 eliminazioni
  1. 12 0
      docs/en/install.md
  2. 13 0
      docs/ja/install.md
  3. 13 0
      docs/ko/install.md
  4. 13 0
      docs/pt/install.md
  5. 13 0
      docs/zh/install.md
  6. 3 0
      tools/run_webui.py

+ 12 - 0
docs/en/install.md

@@ -26,6 +26,18 @@ pip install -e .
 
 uv sync --python 3.12
 ```
+### Intel Arc XPU support
+
+```bash
+conda create -n fish-speech python=3.12
+conda activate fish-speech
+
+conda install libstdcxx -c conda-forge
+
+pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu
+
+pip install -e .
+```
 
 !!! warning
     The `compile` option is not supported on windows and macOS, if you want to run with compile, you need to install trition by yourself.

+ 13 - 0
docs/ja/install.md

@@ -26,5 +26,18 @@ pip install -e .
 uv sync --python 3.12
 ```
 
+### Intel Arc XPU 対応
+
+```bash
+conda create -n fish-speech python=3.12
+conda activate fish-speech
+
+conda install libstdcxx -c conda-forge
+
+pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu
+
+pip install -e .
+```
+
 !!! warning
     `compile` オプションは Windows と macOS でサポートされていません。compile で実行したい場合は、triton を自分でインストールする必要があります。

+ 13 - 0
docs/ko/install.md

@@ -26,5 +26,18 @@ pip install -e .
 uv sync --python 3.12
 ```
 
+### Intel Arc XPU 지원
+
+```bash
+conda create -n fish-speech python=3.12
+conda activate fish-speech
+
+conda install libstdcxx -c conda-forge
+
+pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu
+
+pip install -e .
+```
+
 !!! warning
     `compile` 옵션은 Windows와 macOS에서 지원되지 않습니다. compile로 실행하려면 triton을 직접 설치해야 합니다.

+ 13 - 0
docs/pt/install.md

@@ -26,5 +26,18 @@ pip install -e .
 uv sync --python 3.12
 ```
 
+### Suporte para Intel Arc XPU
+
+```bash
+conda create -n fish-speech python=3.12
+conda activate fish-speech
+
+conda install libstdcxx -c conda-forge
+
+pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu
+
+pip install -e .
+```
+
 !!! warning
     A opção `compile` não é suportada no Windows e macOS, se você quiser executar com compile, precisa instalar o triton por conta própria.

+ 13 - 0
docs/zh/install.md

@@ -26,5 +26,18 @@ pip install -e .
 uv sync --python 3.12
 ```
 
+### Intel Arc XPU 支持
+
+```bash
+conda create -n fish-speech python=3.12
+conda activate fish-speech
+
+conda install libstdcxx -c conda-forge
+
+pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu
+
+pip install -e .
+```
+
 !!! warning
     `compile` 选项在 Windows 和 macOS 上不受支持,如果您想使用 compile 运行,需要自己安装 triton。

+ 3 - 0
tools/run_webui.py

@@ -49,6 +49,9 @@ if __name__ == "__main__":
     if torch.backends.mps.is_available():
         args.device = "mps"
         logger.info("mps is available, running on mps.")
+    elif torch.xpu.is_available():
+        args.device = "xpu"
+        logger.info("XPU is available, running on XPU.")
     elif not torch.cuda.is_available():
         logger.info("CUDA is not available, running on CPU.")
         args.device = "cpu"