ai-team-ori commited on
Commit
3524c13
·
verified ·
1 Parent(s): 768c866

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 = "oriserve/Whisper-Hindi2Hinglish-Prime"
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 = "oriserve/Whisper-Hindi2Hinglish-Prime"
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