Spaces:
Running
Running
Update modules/train_func.py
Browse files- 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-
|
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()
|