Update README.md
Browse files
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 |
-
"
|
| 225 |
torch_dtype=torch_dtype,
|
| 226 |
use_safetensors=True,
|
| 227 |
)
|
| 228 |
model.to(device)
|
| 229 |
|
| 230 |
-
processor = AutoProcessor.from_pretrained("
|
| 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",
|