Update README.md
Browse files
README.md
CHANGED
@@ -16,11 +16,14 @@ Though we only finetune the projector on English and Hindi data, the multilingua
|
|
16 |
You can get started by using huggingface pipeline, as follows:
|
17 |
|
18 |
```
|
|
|
|
|
|
|
19 |
import transformers
|
20 |
import librosa
|
21 |
|
22 |
# load the model pipeline on gpu:0
|
23 |
-
pipe = transformers.pipeline(model='sarvamai/shuka_v1', trust_remote_code=True, device=0)
|
24 |
|
25 |
# get a sample audio
|
26 |
# wget https://huggingface.co/sarvamai/shuka_v1/resolve/main/hi-question.webm
|
|
|
16 |
You can get started by using huggingface pipeline, as follows:
|
17 |
|
18 |
```
|
19 |
+
# install libraries
|
20 |
+
# pip install transformers==4.41.2 peft==0.11.1 librosa==0.10.2
|
21 |
+
|
22 |
import transformers
|
23 |
import librosa
|
24 |
|
25 |
# load the model pipeline on gpu:0
|
26 |
+
pipe = transformers.pipeline(model='sarvamai/shuka_v1', trust_remote_code=True, device=0, torch_dtype='bfloat16')
|
27 |
|
28 |
# get a sample audio
|
29 |
# wget https://huggingface.co/sarvamai/shuka_v1/resolve/main/hi-question.webm
|