1.2

Mateusz-Dera · May 7, 2024 · 7e63b11 · 7e63b11
1 parent f3edd07
commit 7e63b11
Show file tree

Hide file tree

Showing 3 changed files with 100 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -5,18 +5,18 @@ Web UI for WhisperSpeech
 ![Preview](screenshot.png)
 
 ## Info
-[![Version](https://img.shields.io/badge/1.1-version-orange.svg)](https://github.com/Mateusz-Dera/WhisperSpeech-Web-UI/blob/main/README.md)
+[![Version](https://img.shields.io/badge/1.2-version-orange.svg)](https://github.com/Mateusz-Dera/WhisperSpeech-Web-UI/blob/main/README.md)
 
 ### Test platform:
 |Name|Info|
 |:---|:---|
 |CPU|AMD Ryzen 7900X3D (iGPU disabled in BIOS)|
 |GPU|AMD Radeon 7900XTX|
 |RAM|64GB DDR5 6600MHz|
-|Motherboard|ASRock B650E PG Riptide WiFi (2.08.AS01 [Beta])|
+|Motherboard|ASRock B650E PG Riptide WiFi (2.10)|
 |OS|Ubuntu 22.04|
-|Kernel|6.5.0-18-generic|
-|ROCm|6.0|
+|Kernel|6.5.0-28-generic|
+|ROCm|6.1|
 
 ## Instalation:
 1. Install Python 3.11
@@ -34,11 +34,17 @@ export HSA_OVERRIDE_GFX_VERSION=11.0.0
 ```
 
 5. Install requirements
-* For ROCm:
+* ROCm 5.7:
 ```bash
-pip install -r requirements_rocm.txt
+pip install -r requirements_rocm_5.7.txt
 pip install git+https://github.com/ROCmSoftwarePlatform/flash-attention.git@ae7928c5aed53cf6e75cc792baa9126b2abfcf1a
 ```
+
+* ROCm 6.0:
+```bash
+pip install -r requirements_rocm_6.0.txt
+pip install git+https://github.com/ROCmSoftwarePlatform/flash-attention.git@2554f490101742ccdc56620a938f847f61754be6
+```
 
 * CUDA 11.8 (Tested on Ubuntu 23.10):
 ```bash

diff --git a/requirements_rocm.txt → requirements_rocm_5.7.txt b/requirements_rocm.txt → requirements_rocm_5.7.txt
diff --git a/requirements_rocm_6.0.txt b/requirements_rocm_6.0.txt
@@ -0,0 +1,88 @@
+--extra-index-url https://download.pytorch.org/whl/rocm6.0
+aiofiles==23.2.1
+altair==5.2.0
+annotated-types==0.6.0
+anyio==4.2.0
+attrs==23.2.0
+certifi==2022.12.7
+charset-normalizer==2.1.1
+click==8.1.7
+colorama==0.4.6
+contourpy==1.2.0
+cycler==0.12.1
+einops==0.7.0
+encodec==0.1.1
+fastapi==0.109.2
+fastcore==1.5.29
+fastprogress==1.0.3
+ffmpy==0.3.2
+filelock==3.9.0
+fonttools==4.49.0
+fsspec==2024.2.0
+gradio==4.19.1
+gradio_client==0.10.0
+h11==0.14.0
+httpcore==1.0.3
+httpx==0.26.0
+huggingface-hub==0.20.3
+HyperPyYAML==1.2.2
+idna==3.4
+importlib-resources==6.1.1
+Jinja2==3.1.2
+joblib==1.3.2
+jsonschema==4.21.1
+jsonschema-specifications==2023.12.1
+kiwisolver==1.4.5
+markdown-it-py==3.0.0
+MarkupSafe==2.1.3
+matplotlib==3.8.3
+mdurl==0.1.2
+mpmath==1.3.0
+networkx==3.2.1
+ninja==1.11.1.1
+numpy==1.26.3
+orjson==3.9.14
+packaging==23.2
+pandas==2.2.0
+pillow==10.2.0
+pydantic==2.6.1
+pydantic_core==2.16.2
+pydub==0.25.1
+Pygments==2.17.2
+pyparsing==3.1.1
+python-dateutil==2.8.2
+python-multipart==0.0.9
+pytorch-triton-rocm==2.3.0
+pytz==2024.1
+PyYAML==6.0.1
+referencing==0.33.0
+requests==2.28.1
+rich==13.7.0
+rich-argparse==1.4.0
+rpds-py==0.18.0
+ruamel.yaml==0.18.6
+ruamel.yaml.clib==0.2.8
+ruff==0.2.2
+scipy==1.12.0
+semantic-version==2.10.0
+sentencepiece==0.1.99
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.0
+speechbrain==0.5.16
+starlette==0.36.3
+sympy==1.12
+tomlkit==0.12.0
+toolz==0.12.1
+torch==2.3.0+rocm6.0
+torchaudio==2.3.0+rocm6.0
+torchvision==0.18.0+rocm6.0
+tqdm==4.66.2
+typer==0.9.0
+typing_extensions==4.8.0
+tzdata==2024.1
+urllib3==1.26.13
+uvicorn==0.27.1
+vocos==0.1.0
+websockets==11.0.3
+WhisperSpeech==0.6