ai-team-ori
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -163,7 +163,7 @@ device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
|
163 |
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
164 |
|
165 |
# Specify the pre-trained model ID
|
166 |
-
model_id = "
|
167 |
|
168 |
# Load the speech-to-text model with specified configurations
|
169 |
model = AutoModelForSpeechSeq2Seq.from_pretrained(
|
@@ -270,7 +270,7 @@ def save_model(model, save_path):
|
|
270 |
torch.save(pytorch_model, save_path)
|
271 |
|
272 |
# Load Hugging Face model
|
273 |
-
model_id = "
|
274 |
model = AutoModelForSpeechSeq2Seq.from_pretrained(
|
275 |
model_id,
|
276 |
low_cpu_mem_usage=True, # Optimize memory usage
|
|
|
163 |
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
164 |
|
165 |
# Specify the pre-trained model ID
|
166 |
+
model_id = "Oriserve/Whisper-Hindi2Hinglish-Prime"
|
167 |
|
168 |
# Load the speech-to-text model with specified configurations
|
169 |
model = AutoModelForSpeechSeq2Seq.from_pretrained(
|
|
|
270 |
torch.save(pytorch_model, save_path)
|
271 |
|
272 |
# Load Hugging Face model
|
273 |
+
model_id = "Oriserve/Whisper-Hindi2Hinglish-Prime"
|
274 |
model = AutoModelForSpeechSeq2Seq.from_pretrained(
|
275 |
model_id,
|
276 |
low_cpu_mem_usage=True, # Optimize memory usage
|