vishaljoshi24 commited on
Commit
5e1ac07
·
1 Parent(s): 586e73d

Modified LLM to Qwen3

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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/Qwen2.5-0.5B --train_data_dir trl-lib/Capybara --api_key HF_TOKEN" )
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/Qwen2.5-0.5B",
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()