Bootstrapping QwQ works!!

2025-04-02 16:37:17 +00:00
parent b4c08586ee
commit 156d3aa7d3
2 changed files with 4 additions and 2 deletions
--- 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',