Trainings endpoint with cog not working

[centdix]

When i try to do this

curl -X POST "http://localhost:5000/trainings" \
     -H "Content-Type: application/json" \
     -d '{
          "input": {
            "steps": 1000,
            "lora_rank": 16,
            "optimizer": "adamw8bit",
            "batch_size": 1,
            "resolution": "512,768,1024",
            "autocaption": true,
            "input_images": "myzip",
            "trigger_word": "TOK",
            "learning_rate": 0.0004,
            "wandb_project": "flux_train_replicate",
            "wandb_save_interval": 100,
            "caption_dropout_rate": 0.05,
            "cache_latents_to_disk": false,
            "wandb_sample_interval": 100
          }
        }'

I have this error

{"input":{"input_images": "myzip","trigger_word":"TOK","autocaption":true,"autocaption_prefix":null,"autocaption_suffix":null,"steps":1000,"learning_rate":0.0004,"batch_size":1,"resolution":"512,768,1024","lora_rank":16,"caption_dropout_rate":0.05,"optimizer":"adamw8bit","cache_latents_to_disk":false,"layers_to_optimize_regex":null,"gradient_checkpointing":false,"hf_repo_id":null,"hf_token":null,"wandb_api_key":null,"wandb_project":"flux_train_replicate","wandb_run":null,"wandb_entity":null,"wandb_sample_interval":100,"wandb_sample_prompts":null,"wandb_save_interval":100,"skip_training_and_use_pretrained_hf_lora_url":null},"output":null,"id":null,"version":null,"created_at":null,"started_at":"2025-01-30T21:49:21.320825+00:00","completed_at":"2025-01-30T21:49:21.323752+00:00","logs":"Traceback (most recent call last):\nFile \"/root/.pyenv/versions/3.10.15/lib/python3.10/site-packages/cog/server/worker.py\", line 778, in _handle_predict_error\nyield\nFile \"/root/.pyenv/versions/3.10.15/lib/python3.10/site-packages/cog/server/worker.py\", line 634, in _predict\nresult = predict(**payload)\nFile \"/root/.pyenv/versions/3.10.15/lib/python3.10/site-packages/torch/utils/_contextlib.py\", line 116, in decorate_context\nreturn func(*args, **kwargs)\nTypeError: Predictor.predict() got an unexpected keyword argument 'input_images'\n","error":"Predictor.predict() got an unexpected keyword argument 'input_images'","status":"failed","metrics":null}