get the hardware?

README.md

@@ -1,14 +1,12 @@
 ---
 title: Train
-emoji: 👀
+emoji: 🏋️
 colorFrom: pink
 colorTo: purple
 sdk: docker
 pinned: false
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
-
 ```sh
-hfjobs run --flavor a10g-small hf.co/spaces/trl-lib/sft python sft.py --model Qwen/Qwen3-4B-Base --dataset trl-lib/tldr
+hfjobs run --flavor a10g-small hf.co/spaces/trl-lib/train sft --model Qwen/Qwen3-4B-Base --dataset trl-lib/tldr
 ```

demo.py

@@ -0,0 +1,9 @@
+import argparse
+
+parser = argparse.ArgumentParser(description="Demo script for the model.")
+
+parser.add_argument("--model", type=str)
+parser.add_argument("--dataset", type=str)
+
+args = parser.parse_args()
+print(f"Model: {args.model}, Dataset: {args.dataset}")

sft.sh

@@ -1,2 +1,4 @@
 #!/bin/bash
-echo "Hello"
+echo "Hello"
+nvidia-smi --query-gpu=name --format=csv,noheader
+python demo.py "$@"