Spaces:
Paused
Paused
Commit
·
5e1ac07
1
Parent(s):
586e73d
Modified LLM to Qwen3
Browse files- app.py +1 -1
- quickstart.py +1 -1
app.py
CHANGED
@@ -4,7 +4,7 @@ import gradio as gr
|
|
4 |
api_key = os.getenv("HF_TOKEN")
|
5 |
|
6 |
def start_training():
|
7 |
-
os.system("python quickstart.py --model_name_or_path Qwen/
|
8 |
return "Evaluation completed!"
|
9 |
|
10 |
iface = gr.Interface(
|
|
|
4 |
api_key = os.getenv("HF_TOKEN")
|
5 |
|
6 |
def start_training():
|
7 |
+
os.system("python quickstart.py --model_name_or_path Qwen/Qwen3-0.5B --train_data_dir trl-lib/Capybara --api_key HF_TOKEN" )
|
8 |
return "Evaluation completed!"
|
9 |
|
10 |
iface = gr.Interface(
|
quickstart.py
CHANGED
@@ -4,7 +4,7 @@ from datasets import load_dataset
|
|
4 |
dataset = load_dataset("trl-lib/Capybara", split="train")
|
5 |
|
6 |
trainer = SFTTrainer(
|
7 |
-
model="Qwen/
|
8 |
train_dataset=dataset,
|
9 |
)
|
10 |
trainer.train()
|
|
|
4 |
dataset = load_dataset("trl-lib/Capybara", split="train")
|
5 |
|
6 |
trainer = SFTTrainer(
|
7 |
+
model="Qwen/Qwen3-0.6B",
|
8 |
train_dataset=dataset,
|
9 |
)
|
10 |
trainer.train()
|