rahular commited on
Commit
a09d06d
1 Parent(s): a19fbdc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
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