Yehor
/

w2v-bert-uk-v2.1

Automatic Speech Recognition

Model card Files Files and versions Community

Yehor commited on Feb 19

Commit

2bc8cff

·

verified ·

1 Parent(s): 9d20985

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -58,7 +58,7 @@ from transformers import AutoModelForCTC, Wav2Vec2BertProcessor
 # Config
 model_name = 'Yehor/w2v-bert-uk-v2.1'
-device = 'cuda:1' # or cpu
 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 sampling_rate = 16_000

 # Config
 model_name = 'Yehor/w2v-bert-uk-v2.1'
+device = 'cuda:0' # or cpu
 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 sampling_rate = 16_000