Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
loubbrad committed Apr 16, 2024
1 parent 453e3ce commit 63a17e9
Showing 1 changed file with 8 additions and 4 deletions.
12 changes: 8 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -9,19 +9,22 @@ Requires Python 3.11
```
git clone https://github.com/EleutherAI/aria-amt.git
cd aria-amt
pip install -e .```
pip install -e .
```

Download the preliminary model weights:

```
wget https://storage.googleapis.com/aria-checkpoints/amt/small-0.safetensors```
wget https://storage.googleapis.com/aria-checkpoints/amt/small-0.safetensors
```

## Usage

You can download mp3s from youtube using [yt-dlp](https://github.com/yt-dlp/yt-dlp):

```
yt-dlp --audio-format mp3 --extract-audio --no-playlist --audio-quality 0 <youtube-link> -o <save-path>```
yt-dlp --audio-format mp3 --extract-audio --no-playlist --audio-quality 0 <youtube-link> -o <save-path>
```

You can then transcribe using the cli:

Expand All @@ -32,7 +35,8 @@ You can then transcribe using the cli:
-save_dir <path-to-save-dir> \
-bs 1 \
-compile \
-q8```
-q8
```

If you want to do batch transcription, use the `-load_dir` flag and adjust `-bs` accordingly. Compiling may take some time, but provides a significant speedup.

Expand Down

0 comments on commit 63a17e9

Please sign in to comment.