-
Notifications
You must be signed in to change notification settings - Fork 263
Closed
Description
I've run a simple version of the training for flux-dev-lora-trainer.
training = replicate.trainings.create(
version="ostris/flux-dev-lora-trainer:b6af14222e6bd9be257cbc1ea4afda3cd0503e1133083b9d1de0364d8568e6ef",
input= {
"steps": 600,
"lora_rank": 16,
"optimizer": "adamw8bit",
"batch_size": 1,
"resolution": "512,768,1024",
"input_images": S3_URL,
"autocaption_prefix": "a photo of TOP"
"autocaption": False,
"trigger_word": "TOP",
"learning_rate": 0.0005,
"caption_dropout_rate": 0.05,
"cache_latents_to_disk": False,
"gradient_checkpointing": False,
"wandb_project": "flux_train_replicate",
"wandb_save_interval": 100,
"wandb_sample_interval": 100,
},
destination="<ORG>/dusya-cat_new_prefix"
)
As a result, I've got a model that points to the latest version of the flux-dev-lora-trainer (c6e78d2501e8088876e99ef21e4460d0dc121af7a4b786b9a4c2d75c620e300d), not the one I specified.
Am I doing something wrong, or is the API just ignoring the specific model version?
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels
