IPA-Transcription-EN

Running

SanderGi commited on 16 days ago

Commit

f718747

1 Parent(s): c2e60bb

fix typo

Files changed (1) hide show

app/inference.py CHANGED Viewed

@@ -23,7 +23,8 @@ def clear_cache():
     if torch.cuda.is_available():
         torch.cuda.empty_cache()
         torch.cuda.ipc_collect()
-    torch.mps.empty_cache()
 def load_model(model_id, device=DEVICE):

     if torch.cuda.is_available():
         torch.cuda.empty_cache()
         torch.cuda.ipc_collect()
+    if torch.backends.mps.is_available():
+        torch.mps.empty_cache()
 def load_model(model_id, device=DEVICE):