Upload folder using huggingface_hub
Browse files- model_trainer.py +1 -1
model_trainer.py
CHANGED
|
@@ -18,7 +18,7 @@ def authenticate_hf(token: Optional[str]) -> None:
|
|
| 18 |
else:
|
| 19 |
print("Skipping Hugging Face login: HF_TOKEN not set.")
|
| 20 |
|
| 21 |
-
@
|
| 22 |
def load_embedding_model(model_name: str) -> SentenceTransformer:
|
| 23 |
"""Initializes the Sentence Transformer model."""
|
| 24 |
print(f"Loading Sentence Transformer model: {model_name}")
|
|
|
|
| 18 |
else:
|
| 19 |
print("Skipping Hugging Face login: HF_TOKEN not set.")
|
| 20 |
|
| 21 |
+
@spaces.GPU
|
| 22 |
def load_embedding_model(model_name: str) -> SentenceTransformer:
|
| 23 |
"""Initializes the Sentence Transformer model."""
|
| 24 |
print(f"Loading Sentence Transformer model: {model_name}")
|