From 156d3aa7d3693cc64c1e58dd7223b1624d10a43d Mon Sep 17 00:00:00 2001
From: Niels Geens <niels.geens@gmail.com>
Date: Wed, 2 Apr 2025 16:37:17 +0000
Subject: [PATCH] Bootstrapping QwQ works!!

---
 scripts/bootstrap.sh     | 4 +++-
 tools/train/train/qwq.py | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/scripts/bootstrap.sh b/scripts/bootstrap.sh
index 5505436..2568b9f 100755
--- a/scripts/bootstrap.sh
+++ b/scripts/bootstrap.sh
@@ -56,6 +56,7 @@ npm run build
 ln -s "/root/sia/web/dist" "/root/static"
 
 # Install SIA dependencies
+cd /root/sia/desktop
 source "/root/sia/scripts/install.sh"
 
 # Add venvs to path in .bashrc
@@ -73,7 +74,8 @@ source /root/sia/scripts/add_venvs_to_path.sh
 
 # Finetune model
 echo "Finetuning model..."
-train
+train /root/models/bootstrap
+ln -s /root/models/bootstrap /root/models/current
 
 # Start SIA using restart script
 echo "Run restart script..."
diff --git a/tools/train/train/qwq.py b/tools/train/train/qwq.py
index 7db4086..e22152b 100644
--- a/tools/train/train/qwq.py
+++ b/tools/train/train/qwq.py
@@ -18,7 +18,7 @@ from .dataset import Dataset
 
 @dataclass
 class Args:
-    def __init__(self, args: Optional[List[str]]):
+    def __init__(self, args: Optional[List[str]] = None):
         parser = argparse.ArgumentParser(description='Train SIA model using QwQ')
         parser.add_argument(
             '--config',