Update modules/train_func.py

#6
Files changed (1) hide show
  1. modules/train_func.py +1 -1
modules/train_func.py CHANGED
@@ -116,7 +116,7 @@ def build_event_description(id, status, trained_tokens, name=i18n("暂时未知"
116
  def start_training(file_id, suffix, epochs):
117
  openai.api_key = os.getenv("OPENAI_API_KEY")
118
  try:
119
- job = openai.FineTuningJob.create(training_file=file_id, model="gpt-3.5-turbo", suffix=suffix, hyperparameters={"n_epochs": epochs})
120
  return build_event_description(job.id, job.status, job.trained_tokens)
121
  except Exception as e:
122
  traceback.print_exc()
 
116
  def start_training(file_id, suffix, epochs):
117
  openai.api_key = os.getenv("OPENAI_API_KEY")
118
  try:
119
+ job = openai.FineTuningJob.create(training_file=file_id, model="gpt-4o", suffix=suffix, hyperparameters={"n_epochs": epochs})
120
  return build_event_description(job.id, job.status, job.trained_tokens)
121
  except Exception as e:
122
  traceback.print_exc()