bc7ec356 commited on
Commit
767cda1
·
verified ·
1 Parent(s): c09a054

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -221,13 +221,13 @@ device = "cuda:0" if torch.cuda.is_available() else "cpu"
221
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
222
 
223
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
224
- "model-path",
225
  torch_dtype=torch_dtype,
226
  use_safetensors=True,
227
  )
228
  model.to(device)
229
 
230
- processor = AutoProcessor.from_pretrained("model-path")
231
 
232
  pipe = pipeline(
233
  "automatic-speech-recognition",
 
221
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
222
 
223
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
224
+ "bc7ec356/heep-universal",
225
  torch_dtype=torch_dtype,
226
  use_safetensors=True,
227
  )
228
  model.to(device)
229
 
230
+ processor = AutoProcessor.from_pretrained("bc7ec356/heep-universal")
231
 
232
  pipe = pipeline(
233
  "automatic-speech-recognition",