Skip to content

Commit

Permalink
feat: remove sd1.5
Browse files Browse the repository at this point in the history
  • Loading branch information
@jpwallace22
jpwallace22 committed Sep 11, 2024
1 parent ffef84b commit c0c8cac
Show file tree
Hide file tree
Showing 16 changed files with 75 additions and 610 deletions.
145 changes: 51 additions & 94 deletions fern/apis/fine-tuning/openapi/openapi.json
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,13 +2,11 @@
"components": {
"schemas": {
"BaseEngine": {
"description": "Base Engine names supported by asset library.",
"description": "BaseEngine",
"enum": [
"text/llama-2-7b",
"text/llama-2-13b",
"text/llama-2-70b",
"text/llama-3-8b",
"text/llama-3-70b",
"text/codellama-7b",
"text/codellama-13b",
"text/codellama-34b",
Expand Down Expand Up @@ -59,8 +57,7 @@
"discriminator": {
"mapping": {
"lora_tune": "#/components/schemas/LoraTune-Input",
"text_to_speech_latent_tune": "#/components/schemas/TextToSpeechLatentTune",
"text_tune": "#/components/schemas/TextTune"
"text_to_speech_latent_tune": "#/components/schemas/TextToSpeechLatentTune"
},
"propertyName": "tune_type"
},
Expand All @@ -70,9 +67,6 @@
},
{
"$ref": "#/components/schemas/TextToSpeechLatentTune"
},
{
"$ref": "#/components/schemas/TextTune"
}
],
"title": "Details"
Expand All @@ -88,10 +82,8 @@
"type": "object"
},
"HTTPValidationError": {
"description": "HTTP validation error in API.",
"properties": {
"detail": {
"description": "Detail of the http validation error.",
"items": {
"$ref": "#/components/schemas/ValidationError"
},
Expand All @@ -106,7 +98,7 @@
"description": "The list tunes response.",
"properties": {
"data": {
"description": "List of tunes on the account.",
"description": "List of tunes.",
"items": {
"$ref": "#/components/schemas/Tune"
},
Expand All @@ -129,7 +121,7 @@
"type": "object"
},
"LoraTune-Input": {
"description": "A LoRA tune model that includes tune configurations.",
"description": "A LoRA Tune.",
"properties": {
"base_checkpoint": {
"allOf": [
Expand Down Expand Up @@ -188,18 +180,16 @@
"tune_type": {
"const": "lora_tune",
"default": "lora_tune",
"description": "Determines the tune type for a tune request.",
"enum": ["lora_tune"],
"title": "Tune Type",
"type": "string"
"description": "Tune type.",
"title": "Tune Type"
}
},
"required": ["base_checkpoint", "steps", "files", "trigger_words"],
"title": "LoraTune",
"type": "object"
},
"LoraTune-Output": {
"description": "A LoRA tune model that includes tune configurations.",
"description": "A LoRA Tune.",
"properties": {
"base_checkpoint": {
"allOf": [
Expand Down Expand Up @@ -258,10 +248,8 @@
"tune_type": {
"const": "lora_tune",
"default": "lora_tune",
"description": "Determines the tune type for a tune request.",
"enum": ["lora_tune"],
"title": "Tune Type",
"type": "string"
"description": "Tune type.",
"title": "Tune Type"
}
},
"required": ["base_checkpoint", "steps", "files", "trigger_words"],
Expand Down Expand Up @@ -356,57 +344,16 @@
"tune_type": {
"const": "text_to_speech_latent_tune",
"default": "text_to_speech_latent_tune",
"description": "Determines the tune type for a tune request.",
"enum": ["text_to_speech_latent_tune"],
"title": "Tune Type",
"type": "string"
"description": "Tune type.",
"title": "Tune Type"
}
},
"required": ["asset_ids"],
"title": "TextToSpeechLatentTune",
"type": "object"
},
"TextTune": {
"description": "A text generation tune.",
"properties": {
"checkpoint": {
"description": "The checkpoint name or ID used as base model for tuning.",
"title": "Checkpoint",
"type": "string"
},
"epochs": {
"default": 4,
"description": "The number of epochs used for this Text tune.",
"title": "Epochs",
"type": "integer"
},
"learning_rate": {
"default": 0.0001,
"description": "The learning rate used for this Text tune.",
"title": "Learning Rate",
"type": "number"
},
"lora_rank": {
"default": 16,
"description": "The LoRA rank used for this Text tune.",
"title": "Lora Rank",
"type": "integer"
},
"tune_type": {
"const": "text_tune",
"default": "text_tune",
"description": "Determines the tune type for a tune request.",
"enum": ["text_tune"],
"title": "Tune Type",
"type": "string"
}
},
"required": ["checkpoint"],
"title": "TextTune",
"type": "object"
},
"Tune": {
"description": "Tune base class. This class is used to store tune information.",
"description": "A tune base class.",
"properties": {
"created_at": {
"description": "The time this tune task was created.",
Expand All @@ -433,12 +380,11 @@
"type": "string"
},
"details": {
"description": "Tune details. This field is used to store a specific tune type configurations.",
"description": "Tune details.",
"discriminator": {
"mapping": {
"lora_tune": "#/components/schemas/LoraTune-Output",
"text_to_speech_latent_tune": "#/components/schemas/TextToSpeechLatentTune",
"text_tune": "#/components/schemas/TextTune"
"text_to_speech_latent_tune": "#/components/schemas/TextToSpeechLatentTune"
},
"propertyName": "tune_type"
},
Expand All @@ -448,9 +394,6 @@
},
{
"$ref": "#/components/schemas/TextToSpeechLatentTune"
},
{
"$ref": "#/components/schemas/TextTune"
}
],
"title": "Details"
Expand Down Expand Up @@ -621,7 +564,6 @@
"description": "The type of tune requested.",
"enum": [
"lora_tune",
"text_tune",
"text_to_speech_latent_tune",
"text_to_speech_rvc_tune",
"text_to_speech_tortoise_tune"
Expand All @@ -630,10 +572,8 @@
"type": "string"
},
"ValidationError": {
"description": "Validation error in API.",
"properties": {
"loc": {
"description": "Error location in ValidationError.",
"items": {
"anyOf": [
{
Expand All @@ -648,12 +588,10 @@
"type": "array"
},
"msg": {
"description": "Error message in ValidationError.",
"title": "Message",
"type": "string"
},
"type": {
"description": "Error type in ValidationError.",
"title": "Error Type",
"type": "string"
}
Expand All @@ -665,12 +603,32 @@
}
},
"info": {
"description": "Fine-Tuning Service API.",
"title": "fine-tuning-api",
"title": "fine-tuning-api-server",
"version": "0.1.0"
},
"openapi": "3.1.0",
"paths": {
"/health": {
"get": {
"description": "Healthcheck server endpoint.",
"operationId": "health_check_health_get",
"responses": {
"200": {
"content": {
"application/json": {
"schema": {
"title": "Response Health Check Health Get",
"type": "object"
}
}
},
"description": "Successful Response"
}
},
"summary": "Health Check",
"tags": ["tune"]
}
},
"/v1/tune": {
"post": {
"description": "Spawn a tune.",
Expand Down Expand Up @@ -717,12 +675,12 @@
"operationId": "delete_tune_v1_tune__tune_id__delete",
"parameters": [
{
"description": "The ID of the tune to delete.",
"description": "The ID of the tune.",
"in": "path",
"name": "tune_id",
"required": true,
"schema": {
"description": "The ID of the tune to delete.",
"description": "The ID of the tune.",
"title": "Tune Id",
"type": "string"
}
Expand All @@ -733,7 +691,8 @@
"content": {
"application/json": {
"schema": {
"$ref": "#/components/schemas/Tune"
"title": "Response Delete Tune V1 Tune Tune Id Delete",
"type": "object"
}
}
},
Expand All @@ -754,16 +713,16 @@
"tags": ["tune"]
},
"get": {
"description": "Retrieve a specific tune.",
"operationId": "retrieve_tune_v1_tune__tune_id__get",
"description": "Get the specific tune.",
"operationId": "get_tune_v1_tune__tune_id__get",
"parameters": [
{
"description": "The ID of the tune to retrieve.",
"description": "The ID of the tune.",
"in": "path",
"name": "tune_id",
"required": true,
"schema": {
"description": "The ID of the tune to retrieve.",
"description": "The ID of the tune.",
"title": "Tune Id",
"type": "string"
}
Expand Down Expand Up @@ -791,7 +750,7 @@
"description": "Validation Error"
}
},
"summary": "Retrieve Tune",
"summary": "Get Tune",
"tags": ["tune"]
}
},
Expand Down Expand Up @@ -851,6 +810,7 @@
"schema": {
"default": 0,
"description": "Offset into the results.",
"gte": 0,
"title": "Offset",
"type": "integer"
}
Expand All @@ -863,6 +823,8 @@
"schema": {
"default": 100,
"description": "The max number of results to be shown (limit 100).",
"gte": 1,
"lte": 100,
"title": "Limit",
"type": "integer"
}
Expand Down Expand Up @@ -944,7 +906,7 @@
}
},
{
"description": "The engine type for listing tunes.",
"description": "The engine type.",
"in": "query",
"name": "engine",
"required": false,
Expand All @@ -957,7 +919,7 @@
"type": "null"
}
],
"description": "The engine type for listing tunes.",
"description": "The engine type.",
"title": "Engine"
}
}
Expand Down Expand Up @@ -988,10 +950,5 @@
"tags": ["tune"]
}
}
},
"servers": [
{
"url": "https://api.octoai.cloud"
}
]
}
}
Loading

0 comments on commit c0c8cac

Please sign in to comment.