fix: use any for n_epochs (#499)
* fix: use custom marshaler for n_epochs * chore: use any for n_epochs
This commit is contained in:
@@ -24,7 +24,7 @@ type FineTuningJob struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type Hyperparameters struct {
|
type Hyperparameters struct {
|
||||||
Epochs int `json:"n_epochs"`
|
Epochs any `json:"n_epochs,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type FineTuningJobRequest struct {
|
type FineTuningJobRequest struct {
|
||||||
|
|||||||
@@ -21,8 +21,23 @@ func TestFineTuningJob(t *testing.T) {
|
|||||||
server.RegisterHandler(
|
server.RegisterHandler(
|
||||||
"/v1/fine_tuning/jobs",
|
"/v1/fine_tuning/jobs",
|
||||||
func(w http.ResponseWriter, r *http.Request) {
|
func(w http.ResponseWriter, r *http.Request) {
|
||||||
var resBytes []byte
|
resBytes, _ := json.Marshal(FineTuningJob{
|
||||||
resBytes, _ = json.Marshal(FineTuningJob{})
|
Object: "fine_tuning.job",
|
||||||
|
ID: testFineTuninigJobID,
|
||||||
|
Model: "davinci-002",
|
||||||
|
CreatedAt: 1692661014,
|
||||||
|
FinishedAt: 1692661190,
|
||||||
|
FineTunedModel: "ft:davinci-002:my-org:custom_suffix:7q8mpxmy",
|
||||||
|
OrganizationID: "org-123",
|
||||||
|
ResultFiles: []string{"file-abc123"},
|
||||||
|
Status: "succeeded",
|
||||||
|
ValidationFile: "",
|
||||||
|
TrainingFile: "file-abc123",
|
||||||
|
Hyperparameters: Hyperparameters{
|
||||||
|
Epochs: "auto",
|
||||||
|
},
|
||||||
|
TrainedTokens: 5768,
|
||||||
|
})
|
||||||
fmt.Fprintln(w, string(resBytes))
|
fmt.Fprintln(w, string(resBytes))
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user