Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: remove sd1.5 #104

Merged
merged 2 commits into from
Sep 11, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
145 changes: 51 additions & 94 deletions fern/apis/fine-tuning/openapi/openapi.json
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"
}
]
}
}
14 changes: 0 additions & 14 deletions fern/apis/image-gen/openapi/openapi-overrides.yml
Original file line number Diff line number Diff line change
Expand Up @@ -9,27 +9,13 @@ paths:
servers:
- url: https://image.octoai.run
x-name: ImageGen
/generate/controlnet-sd15:
post:
summary: "Generate ControlNet SD1.5"
x-fern-sdk-method-name: generateControlnetSd15
servers:
- url: https://image.octoai.run
x-name: ImageGen
/generate/sdxl:
post:
summary: "Generate SDXL"
x-fern-sdk-method-name: generateSdxl
servers:
- url: https://image.octoai.run
x-name: ImageGen
/generate/sd:
post:
summary: "Generate SD1.5"
x-fern-sdk-method-name: generateSd
servers:
- url: https://image.octoai.run
x-name: ImageGen
/generate/sd3:
post:
summary: "Generate SD3"
Expand Down
Loading
Loading