565 lines
28 KiB
JSON
565 lines
28 KiB
JSON
{
|
|
"url": "https://replicate.com/lucataco/magnet",
|
|
"owner": "lucataco",
|
|
"name": "magnet",
|
|
"description": "MAGNeT: Masked Audio Generation using a Single Non-Autoregressive Transformer",
|
|
"visibility": "public",
|
|
"github_url": "https://github.com/lucataco/audiocraft",
|
|
"paper_url": "https://arxiv.org/abs/2401.04577",
|
|
"license_url": "https://github.com/facebookresearch/audiocraft?tab=MIT-1-ov-file#readme",
|
|
"run_count": 0,
|
|
"cover_image_url": "https://tjzk.replicate.delivery/models_models_cover_image/b958c60f-aa4a-44ac-b251-34317cd96245/magnet.jpeg",
|
|
"default_example": {
|
|
"id": "6usmrb3bog2on4b52tcnpt74oa",
|
|
"model": "lucataco/magnet",
|
|
"version": "e8e2ecd4a1dabb58924aa8300b668290cafae166dd36baf65dad9875877de50e",
|
|
"status": "succeeded",
|
|
"input": {
|
|
"model": "facebook/magnet-small-10secs",
|
|
"top_p": 0.9,
|
|
"prompt": "80s electronic track with melodic synthesizers, catchy beat and groovy bass",
|
|
"max_cfg": 10,
|
|
"min_cfg": 1,
|
|
"span_score": "prod-stride1",
|
|
"variations": 3,
|
|
"temperature": 3,
|
|
"decoding_steps_stage_1": 20,
|
|
"decoding_steps_stage_2": 10,
|
|
"decoding_steps_stage_3": 10,
|
|
"decoding_steps_stage_4": 10
|
|
},
|
|
"output": [
|
|
"https://replicate.delivery/pbxt/A09884lLfO3qWidseZLbU90KjWL3eteHOcSafheMaucmacXjE/0.wav",
|
|
"https://replicate.delivery/pbxt/dG0LoAeZINxcf0xueWgwgNHyF2tQv3Lvo7S7HfEew2JbNurRC/1.wav",
|
|
"https://replicate.delivery/pbxt/K55hoe139wzvBqXV4pkfDMfqbqRMaYVnJ6O8M1efLbteacXjE/2.wav"
|
|
],
|
|
"logs": "/root/.pyenv/versions/3.10.13/lib/python3.10/site-packages/torch/nn/utils/weight_norm.py:30: UserWarning: torch.nn.utils.weight_norm is deprecated in favor of torch.nn.utils.parametrizations.weight_norm.\nwarnings.warn(\"torch.nn.utils.weight_norm is deprecated in favor of torch.nn.utils.parametrizations.weight_norm.\")\nspiece.model: 0%| | 0.00/792k [00:00<?, ?B/s]\nspiece.model: 100%|██████████| 792k/792k [00:00<00:00, 31.9MB/s]\ntokenizer.json: 0%| | 0.00/1.39M [00:00<?, ?B/s]\ntokenizer.json: 100%|██████████| 1.39M/1.39M [00:00<00:00, 24.7MB/s]\nconfig.json: 0%| | 0.00/1.21k [00:00<?, ?B/s]\nconfig.json: 100%|██████████| 1.21k/1.21k [00:00<00:00, 10.4MB/s]\nmodel.safetensors: 0%| | 0.00/892M [00:00<?, ?B/s]\nmodel.safetensors: 4%|▎ | 31.5M/892M [00:00<00:05, 171MB/s]\nmodel.safetensors: 7%|▋ | 62.9M/892M [00:00<00:04, 193MB/s]\nmodel.safetensors: 11%|█ | 94.4M/892M [00:00<00:03, 208MB/s]\nmodel.safetensors: 14%|█▍ | 126M/892M [00:00<00:03, 231MB/s] \nmodel.safetensors: 18%|█▊ | 157M/892M [00:00<00:03, 213MB/s]\nmodel.safetensors: 22%|██▏ | 199M/892M [00:00<00:03, 227MB/s]\nmodel.safetensors: 26%|██▌ | 231M/892M [00:01<00:02, 232MB/s]\nmodel.safetensors: 29%|██▉ | 262M/892M [00:01<00:02, 234MB/s]\nmodel.safetensors: 33%|███▎ | 294M/892M [00:01<00:02, 235MB/s]\nmodel.safetensors: 36%|███▋ | 325M/892M [00:01<00:02, 239MB/s]\nmodel.safetensors: 40%|███▉ | 357M/892M [00:01<00:02, 233MB/s]\nmodel.safetensors: 44%|████▎ | 388M/892M [00:01<00:02, 233MB/s]\nmodel.safetensors: 47%|████▋ | 419M/892M [00:01<00:02, 235MB/s]\nmodel.safetensors: 51%|█████ | 451M/892M [00:01<00:01, 236MB/s]\nmodel.safetensors: 54%|█████▍ | 482M/892M [00:02<00:01, 227MB/s]\nmodel.safetensors: 58%|█████▊ | 514M/892M [00:02<00:01, 229MB/s]\nmodel.safetensors: 61%|██████ | 545M/892M [00:02<00:01, 228MB/s]\nmodel.safetensors: 65%|██████▍ | 577M/892M [00:02<00:01, 230MB/s]\nmodel.safetensors: 68%|██████▊ | 608M/892M [00:02<00:01, 238MB/s]\nmodel.safetensors: 72%|███████▏ | 640M/892M [00:02<00:01, 246MB/s]\nmodel.safetensors: 75%|███████▌ | 671M/892M [00:02<00:00, 245MB/s]\nmodel.safetensors: 79%|███████▉ | 703M/892M [00:03<00:00, 244MB/s]\nmodel.safetensors: 82%|████████▏ | 734M/892M [00:03<00:00, 242MB/s]\nmodel.safetensors: 86%|████████▌ | 765M/892M [00:03<00:00, 175MB/s]\nmodel.safetensors: 88%|████████▊ | 786M/892M [00:03<00:00, 155MB/s]\nmodel.safetensors: 92%|█████████▏| 818M/892M [00:03<00:00, 175MB/s]\nmodel.safetensors: 95%|█████████▌| 849M/892M [00:03<00:00, 190MB/s]\nmodel.safetensors: 99%|█████████▉| 881M/892M [00:04<00:00, 200MB/s]\nmodel.safetensors: 100%|██████████| 892M/892M [00:04<00:00, 217MB/s]",
|
|
"error": null,
|
|
"metrics": {
|
|
"predict_time": 14.638626,
|
|
"total_time": 110.299311
|
|
},
|
|
"created_at": "2024-01-17T19:26:53.835883Z",
|
|
"started_at": "2024-01-17T19:28:29.496568Z",
|
|
"completed_at": "2024-01-17T19:28:44.135194Z",
|
|
"urls": {
|
|
"get": "https://api.replicate.com/v1/predictions/6usmrb3bog2on4b52tcnpt74oa",
|
|
"cancel": "https://api.replicate.com/v1/predictions/6usmrb3bog2on4b52tcnpt74oa/cancel"
|
|
}
|
|
},
|
|
"latest_version": {
|
|
"id": "e8e2ecd4a1dabb58924aa8300b668290cafae166dd36baf65dad9875877de50e",
|
|
"created_at": "2024-01-17T19:21:50.415392+00:00",
|
|
"cog_version": "0.8.6",
|
|
"openapi_schema": {
|
|
"info": {
|
|
"title": "Cog",
|
|
"version": "0.1.0"
|
|
},
|
|
"paths": {
|
|
"/": {
|
|
"get": {
|
|
"summary": "Root",
|
|
"responses": {
|
|
"200": {
|
|
"content": {
|
|
"application/json": {
|
|
"schema": {
|
|
"title": "Response Root Get"
|
|
}
|
|
}
|
|
},
|
|
"description": "Successful Response"
|
|
}
|
|
},
|
|
"operationId": "root__get"
|
|
}
|
|
},
|
|
"/shutdown": {
|
|
"post": {
|
|
"summary": "Start Shutdown",
|
|
"responses": {
|
|
"200": {
|
|
"content": {
|
|
"application/json": {
|
|
"schema": {
|
|
"title": "Response Start Shutdown Shutdown Post"
|
|
}
|
|
}
|
|
},
|
|
"description": "Successful Response"
|
|
}
|
|
},
|
|
"operationId": "start_shutdown_shutdown_post"
|
|
}
|
|
},
|
|
"/predictions": {
|
|
"post": {
|
|
"summary": "Predict",
|
|
"responses": {
|
|
"200": {
|
|
"content": {
|
|
"application/json": {
|
|
"schema": {
|
|
"$ref": "#/components/schemas/PredictionResponse"
|
|
}
|
|
}
|
|
},
|
|
"description": "Successful Response"
|
|
},
|
|
"422": {
|
|
"content": {
|
|
"application/json": {
|
|
"schema": {
|
|
"$ref": "#/components/schemas/HTTPValidationError"
|
|
}
|
|
}
|
|
},
|
|
"description": "Validation Error"
|
|
}
|
|
},
|
|
"parameters": [
|
|
{
|
|
"in": "header",
|
|
"name": "prefer",
|
|
"schema": {
|
|
"type": "string",
|
|
"title": "Prefer"
|
|
},
|
|
"required": false
|
|
}
|
|
],
|
|
"description": "Run a single prediction on the model",
|
|
"operationId": "predict_predictions_post",
|
|
"requestBody": {
|
|
"content": {
|
|
"application/json": {
|
|
"schema": {
|
|
"$ref": "#/components/schemas/PredictionRequest"
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
},
|
|
"/health-check": {
|
|
"get": {
|
|
"summary": "Healthcheck",
|
|
"responses": {
|
|
"200": {
|
|
"content": {
|
|
"application/json": {
|
|
"schema": {
|
|
"title": "Response Healthcheck Health Check Get"
|
|
}
|
|
}
|
|
},
|
|
"description": "Successful Response"
|
|
}
|
|
},
|
|
"operationId": "healthcheck_health_check_get"
|
|
}
|
|
},
|
|
"/predictions/{prediction_id}": {
|
|
"put": {
|
|
"summary": "Predict Idempotent",
|
|
"responses": {
|
|
"200": {
|
|
"content": {
|
|
"application/json": {
|
|
"schema": {
|
|
"$ref": "#/components/schemas/PredictionResponse"
|
|
}
|
|
}
|
|
},
|
|
"description": "Successful Response"
|
|
},
|
|
"422": {
|
|
"content": {
|
|
"application/json": {
|
|
"schema": {
|
|
"$ref": "#/components/schemas/HTTPValidationError"
|
|
}
|
|
}
|
|
},
|
|
"description": "Validation Error"
|
|
}
|
|
},
|
|
"parameters": [
|
|
{
|
|
"in": "path",
|
|
"name": "prediction_id",
|
|
"schema": {
|
|
"type": "string",
|
|
"title": "Prediction ID"
|
|
},
|
|
"required": true
|
|
},
|
|
{
|
|
"in": "header",
|
|
"name": "prefer",
|
|
"schema": {
|
|
"type": "string",
|
|
"title": "Prefer"
|
|
},
|
|
"required": false
|
|
}
|
|
],
|
|
"description": "Run a single prediction on the model (idempotent creation).",
|
|
"operationId": "predict_idempotent_predictions__prediction_id__put",
|
|
"requestBody": {
|
|
"content": {
|
|
"application/json": {
|
|
"schema": {
|
|
"allOf": [
|
|
{
|
|
"$ref": "#/components/schemas/PredictionRequest"
|
|
}
|
|
],
|
|
"title": "Prediction Request"
|
|
}
|
|
}
|
|
},
|
|
"required": true
|
|
}
|
|
}
|
|
},
|
|
"/predictions/{prediction_id}/cancel": {
|
|
"post": {
|
|
"summary": "Cancel",
|
|
"responses": {
|
|
"200": {
|
|
"content": {
|
|
"application/json": {
|
|
"schema": {
|
|
"title": "Response Cancel Predictions Prediction Id Cancel Post"
|
|
}
|
|
}
|
|
},
|
|
"description": "Successful Response"
|
|
},
|
|
"422": {
|
|
"content": {
|
|
"application/json": {
|
|
"schema": {
|
|
"$ref": "#/components/schemas/HTTPValidationError"
|
|
}
|
|
}
|
|
},
|
|
"description": "Validation Error"
|
|
}
|
|
},
|
|
"parameters": [
|
|
{
|
|
"in": "path",
|
|
"name": "prediction_id",
|
|
"schema": {
|
|
"type": "string",
|
|
"title": "Prediction ID"
|
|
},
|
|
"required": true
|
|
}
|
|
],
|
|
"description": "Cancel a running prediction",
|
|
"operationId": "cancel_predictions__prediction_id__cancel_post"
|
|
}
|
|
}
|
|
},
|
|
"openapi": "3.0.2",
|
|
"components": {
|
|
"schemas": {
|
|
"Input": {
|
|
"type": "object",
|
|
"title": "Input",
|
|
"properties": {
|
|
"model": {
|
|
"allOf": [
|
|
{
|
|
"$ref": "#/components/schemas/model"
|
|
}
|
|
],
|
|
"default": "facebook/magnet-small-10secs",
|
|
"x-order": 1,
|
|
"description": "Model to use"
|
|
},
|
|
"top_p": {
|
|
"type": "number",
|
|
"title": "Top P",
|
|
"default": 0.9,
|
|
"maximum": 1,
|
|
"minimum": 0,
|
|
"x-order": 5,
|
|
"description": "Top p for sampling"
|
|
},
|
|
"prompt": {
|
|
"type": "string",
|
|
"title": "Prompt",
|
|
"default": "80s electronic track with melodic synthesizers, catchy beat and groovy bass",
|
|
"x-order": 0,
|
|
"description": "Input Text"
|
|
},
|
|
"max_cfg": {
|
|
"type": "number",
|
|
"title": "Max Cfg",
|
|
"default": 10,
|
|
"x-order": 6,
|
|
"description": "Max CFG coefficient"
|
|
},
|
|
"min_cfg": {
|
|
"type": "number",
|
|
"title": "Min Cfg",
|
|
"default": 1,
|
|
"x-order": 7,
|
|
"description": "Min CFG coefficient"
|
|
},
|
|
"span_score": {
|
|
"allOf": [
|
|
{
|
|
"$ref": "#/components/schemas/span_score"
|
|
}
|
|
],
|
|
"default": "prod-stride1",
|
|
"x-order": 3
|
|
},
|
|
"variations": {
|
|
"type": "integer",
|
|
"title": "Variations",
|
|
"default": 3,
|
|
"maximum": 4,
|
|
"minimum": 1,
|
|
"x-order": 2,
|
|
"description": "Number of variations to generate"
|
|
},
|
|
"temperature": {
|
|
"type": "number",
|
|
"title": "Temperature",
|
|
"default": 3,
|
|
"x-order": 4,
|
|
"description": "Temperature for sampling"
|
|
},
|
|
"decoding_steps_stage_1": {
|
|
"type": "integer",
|
|
"title": "Decoding Steps Stage 1",
|
|
"default": 20,
|
|
"x-order": 8,
|
|
"description": "Number of decoding steps for stage 1"
|
|
},
|
|
"decoding_steps_stage_2": {
|
|
"type": "integer",
|
|
"title": "Decoding Steps Stage 2",
|
|
"default": 10,
|
|
"x-order": 9,
|
|
"description": "Number of decoding steps for stage 2"
|
|
},
|
|
"decoding_steps_stage_3": {
|
|
"type": "integer",
|
|
"title": "Decoding Steps Stage 3",
|
|
"default": 10,
|
|
"x-order": 10,
|
|
"description": "Number of decoding steps for stage 3"
|
|
},
|
|
"decoding_steps_stage_4": {
|
|
"type": "integer",
|
|
"title": "Decoding Steps Stage 4",
|
|
"default": 10,
|
|
"x-order": 11,
|
|
"description": "Number of decoding steps for stage 4"
|
|
}
|
|
}
|
|
},
|
|
"model": {
|
|
"enum": [
|
|
"facebook/magnet-small-10secs",
|
|
"facebook/magnet-medium-10secs",
|
|
"facebook/magnet-small-30secs",
|
|
"facebook/magnet-medium-30secs",
|
|
"facebook/audio-magnet-small",
|
|
"facebook/audio-magnet-medium"
|
|
],
|
|
"type": "string",
|
|
"title": "model",
|
|
"description": "An enumeration."
|
|
},
|
|
"Output": {
|
|
"type": "array",
|
|
"items": {
|
|
"type": "string",
|
|
"format": "uri"
|
|
},
|
|
"title": "Output"
|
|
},
|
|
"Status": {
|
|
"enum": [
|
|
"starting",
|
|
"processing",
|
|
"succeeded",
|
|
"canceled",
|
|
"failed"
|
|
],
|
|
"type": "string",
|
|
"title": "Status",
|
|
"description": "An enumeration."
|
|
},
|
|
"span_score": {
|
|
"enum": [
|
|
"max-nonoverlap",
|
|
"prod-stride1"
|
|
],
|
|
"type": "string",
|
|
"title": "span_score",
|
|
"description": "An enumeration."
|
|
},
|
|
"WebhookEvent": {
|
|
"enum": [
|
|
"start",
|
|
"output",
|
|
"logs",
|
|
"completed"
|
|
],
|
|
"type": "string",
|
|
"title": "WebhookEvent",
|
|
"description": "An enumeration."
|
|
},
|
|
"ValidationError": {
|
|
"type": "object",
|
|
"title": "ValidationError",
|
|
"required": [
|
|
"loc",
|
|
"msg",
|
|
"type"
|
|
],
|
|
"properties": {
|
|
"loc": {
|
|
"type": "array",
|
|
"items": {
|
|
"anyOf": [
|
|
{
|
|
"type": "string"
|
|
},
|
|
{
|
|
"type": "integer"
|
|
}
|
|
]
|
|
},
|
|
"title": "Location"
|
|
},
|
|
"msg": {
|
|
"type": "string",
|
|
"title": "Message"
|
|
},
|
|
"type": {
|
|
"type": "string",
|
|
"title": "Error Type"
|
|
}
|
|
}
|
|
},
|
|
"PredictionRequest": {
|
|
"type": "object",
|
|
"title": "PredictionRequest",
|
|
"properties": {
|
|
"id": {
|
|
"type": "string",
|
|
"title": "Id"
|
|
},
|
|
"input": {
|
|
"$ref": "#/components/schemas/Input"
|
|
},
|
|
"webhook": {
|
|
"type": "string",
|
|
"title": "Webhook",
|
|
"format": "uri",
|
|
"maxLength": 65536,
|
|
"minLength": 1
|
|
},
|
|
"created_at": {
|
|
"type": "string",
|
|
"title": "Created At",
|
|
"format": "date-time"
|
|
},
|
|
"output_file_prefix": {
|
|
"type": "string",
|
|
"title": "Output File Prefix"
|
|
},
|
|
"webhook_events_filter": {
|
|
"type": "array",
|
|
"items": {
|
|
"$ref": "#/components/schemas/WebhookEvent"
|
|
},
|
|
"default": [
|
|
"start",
|
|
"output",
|
|
"logs",
|
|
"completed"
|
|
]
|
|
}
|
|
}
|
|
},
|
|
"PredictionResponse": {
|
|
"type": "object",
|
|
"title": "PredictionResponse",
|
|
"properties": {
|
|
"id": {
|
|
"type": "string",
|
|
"title": "Id"
|
|
},
|
|
"logs": {
|
|
"type": "string",
|
|
"title": "Logs",
|
|
"default": ""
|
|
},
|
|
"error": {
|
|
"type": "string",
|
|
"title": "Error"
|
|
},
|
|
"input": {
|
|
"$ref": "#/components/schemas/Input"
|
|
},
|
|
"output": {
|
|
"$ref": "#/components/schemas/Output"
|
|
},
|
|
"status": {
|
|
"$ref": "#/components/schemas/Status"
|
|
},
|
|
"metrics": {
|
|
"type": "object",
|
|
"title": "Metrics"
|
|
},
|
|
"version": {
|
|
"type": "string",
|
|
"title": "Version"
|
|
},
|
|
"created_at": {
|
|
"type": "string",
|
|
"title": "Created At",
|
|
"format": "date-time"
|
|
},
|
|
"started_at": {
|
|
"type": "string",
|
|
"title": "Started At",
|
|
"format": "date-time"
|
|
},
|
|
"completed_at": {
|
|
"type": "string",
|
|
"title": "Completed At",
|
|
"format": "date-time"
|
|
}
|
|
}
|
|
},
|
|
"HTTPValidationError": {
|
|
"type": "object",
|
|
"title": "HTTPValidationError",
|
|
"properties": {
|
|
"detail": {
|
|
"type": "array",
|
|
"items": {
|
|
"$ref": "#/components/schemas/ValidationError"
|
|
},
|
|
"title": "Detail"
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} |