Import hf_hub_download to download the model during inference.

collabora · May 7, 2024 · 104fe9c · 104fe9c
1 parent e94cefb
commit 104fe9c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/whisperspeech/inference.py b/whisperspeech/inference.py
@@ -6,6 +6,7 @@
 # %% ../nbs/D. Common inference utilities.ipynb 1
 import torch
 import torch.nn.functional as F
+from huggingface_hub import hf_hub_download
 
 from contextlib import nullcontext