Upload train.py with huggingface_hub

train.py

@@ -2,13 +2,23 @@
 # requires-python = ">=3.10"
 # dependencies = [
 #     "sentence-transformers>=3.0.0",
-#     "torch>=2.0.0",
+#     "torch",
 #     "transformers>=4.40.0",
 #     "numpy",
 #     "einops",
 #     "datasets",
 #     "accelerate>=1.1.0",
 # ]
+#
+# [tool.uv.sources]
+# torch = [
+#     { index = "pytorch-cu124", marker = "sys_platform == 'linux'" },
+# ]
+#
+# [[tool.uv.index]]
+# name = "pytorch-cu124"
+# url = "https://download.pytorch.org/whl/cu124"
+# explicit = true
 # ///
 """
 Self-contained automotive embedding fine-tuning + evaluation.
@@ -548,7 +558,11 @@ def evaluate_model(model: SentenceTransformer, corpus_names: list[str]) -> dict:
 # ============================================================================
 
 def train():
+    import torch
     print(f"=== Automotive Embedding Fine-Tuning ===")
+    print(f"CUDA available: {torch.cuda.is_available()}")
+    if torch.cuda.is_available():
+        print(f"GPU: {torch.cuda.get_device_name(0)}")
     print(f"Base model: {BASE_MODEL}")
     print(f"Epochs: {EPOCHS}, Batch size: {BATCH_SIZE}, LR: {LEARNING_RATE}")
     print(f"Matryoshka dims: {MATRYOSHKA_DIMS}")