bebechien commited on
Commit
2382294
·
verified ·
1 Parent(s): a427de5

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. 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
- @space.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}")
 
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}")