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