Update init_DolphinGR00T_zero.py
Browse files
init_DolphinGR00T_zero.py
CHANGED
@@ -309,5 +309,5 @@ def main(device: str, out_dir: str):
|
|
309 |
if __name__ == "__main__":
|
310 |
ap = argparse.ArgumentParser()
|
311 |
ap.add_argument("--device", default="cpu")
|
312 |
-
ap.add_argument("--out_dir", default="
|
313 |
args = ap.parse_args(); main(args.device, args.out_dir)
|
|
|
309 |
if __name__ == "__main__":
|
310 |
ap = argparse.ArgumentParser()
|
311 |
ap.add_argument("--device", default="cpu")
|
312 |
+
ap.add_argument("--out_dir", default="DolphinGR00T-N1.5-3B-Zero")
|
313 |
args = ap.parse_args(); main(args.device, args.out_dir)
|