mira/unsloth-train-scripts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: up

Browse Source
This commit is contained in:
Mira
2025-02-14 02:13:40 +06:00
parent f013e8efe6
commit 45f45f4bdb
3 changed files with 37 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

@@ -10,9 +10,7 @@ def parse_args():
# wandb args
wandb_group = parser.add_argument_group("Weights & Biases")
wandb_group.add_argument(
"--wandb_project", type=str, default="lora-training", help="WandB project name"
)
wandb_group.add_argument("--wandb_project", type=str, help="WandB project name")
wandb_group.add_argument("--wandb_name", type=str, help="WandB run name")
wandb_group.add_argument("--wandb_entity", type=str, help="WandB entity/username")
wandb_group.add_argument(
@@ -42,7 +40,7 @@ def main():
try:
wandb_config = WandBConfig(
enabled=args.wandb_project is not None,
project=args.wandb_project or "lora-training",
project=args.wandb_project,
name=args.wandb_name,
entity=args.wandb_entity,
tags=args.wandb_tags,