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

removing prediction test cases #66

Merged
merged 1 commit into from
Jan 24, 2025
Merged
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
105 changes: 0 additions & 105 deletions cog-safe-push.yaml.tpl
Original file line number Diff line number Diff line change
@@ -1,110 +1,5 @@
model: ostris/flux-dev-lora-trainer
test_model: replicate-internal/test-flux-fine-tuner
predict:
compare_outputs: false
predict_timeout: 300
test_cases:

# basic
- inputs:
prompt: PLSM style, an image of a tiger, made of yellow, red, and purple goo plasma
num_outputs: 1
lora_scale: 1.1
num_inference_steps: 28
guidance_scale: 1.5
seed: 5259
output_format: jpg
replicate_weights: https://replicate.delivery/yhqm/MHJmIF7zlLKXPNN9LKN5yxYUuC7SyzKjMBtqeVwUGdPCpqqJA/trained_model.tar
match_url: https://storage.googleapis.com/replicate-test-flux-finer-tuner/test-outputs/base.jpg

# img2img and custom size
- inputs:
prompt: "PLSM style, an image of a formula one car, made of yellow, red, and purple goo plasma"
num_outputs: 1
lora_scale: 1.0
aspect_ratio: "custom"
width: 500
height: 256
num_inference_steps: 20
model: "dev"
guidance_scale: 1.5
replicate_weights: "https://replicate.delivery/yhqm/MHJmIF7zlLKXPNN9LKN5yxYUuC7SyzKjMBtqeVwUGdPCpqqJA/trained_model.tar"
output_format: "jpg"
output_quality: 90
image: "https://storage.googleapis.com/cog-safe-push-public/fast-car.jpg"
prompt_strength: 0.9
seed: 12919
match_url: https://replicate.delivery/yhqm/zekZzRptTet51UvOUHUH0fnT04taL9seSvRQuP0p7fPnvoIcC/out-0.jpg

# inpainting
- inputs:
prompt: "PLSM style, an image of a tiger, made of yellow, red, and purple goo plasma"
num_outputs: 1
lora_scale: 0.8
aspect_ratio: custom
width: 500
height: 256
num_inference_steps: 20
model: "dev"
guidance_scale: 1.5
replicate_weights: "https://replicate.delivery/yhqm/MHJmIF7zlLKXPNN9LKN5yxYUuC7SyzKjMBtqeVwUGdPCpqqJA/trained_model.tar"
output_format: "jpg"
output_quality: 90
prompt_strength: 1.0
image: "https://replicate.delivery/pbxt/LbXcDNOFtOMiBChe9QHHbeNJ5KcoTNfWEGmIxswhaexvXWEs/Screenshot%202024-09-11%20at%2014.36.07.png"
mask: "https://replicate.delivery/pbxt/LbXcE8bXqlTqPWIDb3SFFuJzIUMKSUscnqF97XVv0K2coEwr/Screenshot%202024-09-11%20at%2014.37.41.png"
seed: 64090
match_url: https://storage.googleapis.com/replicate-test-flux-finer-tuner/test-outputs/inpaint.jpg

# aspect ratio
- inputs:
prompt: PLSM style, an image of a tiger, made of yellow, red, and purple goo plasma
aspect_ratio: "16:9"
num_outputs: 1
lora_scale: 1.1
num_inference_steps: 28
guidance_scale: 1.5
seed: 16726
output_format: jpg
replicate_weights: https://replicate.delivery/yhqm/MHJmIF7zlLKXPNN9LKN5yxYUuC7SyzKjMBtqeVwUGdPCpqqJA/trained_model.tar
match_url: https://storage.googleapis.com/replicate-test-flux-finer-tuner/test-outputs/aspect-ratio.jpg

# schnell
- inputs:
prompt: PLSM style, an image of a tiger, made of yellow, red, and purple goo plasma
num_outputs: 1
lora_scale: 1.1
num_inference_steps: 4
model: schnell
guidance_scale: 1.5
seed: 16726
output_format: jpg
replicate_weights: https://replicate.delivery/yhqm/MHJmIF7zlLKXPNN9LKN5yxYUuC7SyzKjMBtqeVwUGdPCpqqJA/trained_model.tar
match_url: https://replicate.delivery/yhqm/rQhMIg6dhLYVJ5gXWfZI4Ik1DF9cGV7pm8UN0DKmpWDfJGhTA/out-0.jpg

# extra lora
- inputs:
prompt: PLSM style, an image of a man ANDRS, made of yellow, red, and purple goo plasma
num_outputs: 1
lora_scale: 0.8
num_inference_steps: 20
model: dev
guidance_scale: 1.5
seed: 413
extra_lora: andreasjansson/flux-me
extra_lora_scale: 1.1
replicate_weights: https://replicate.delivery/yhqm/MHJmIF7zlLKXPNN9LKN5yxYUuC7SyzKjMBtqeVwUGdPCpqqJA/trained_model.tar
output_format: jpg
output_quality: 90
match_url: https://storage.googleapis.com/replicate-test-flux-finer-tuner/test-outputs/extra-lora.jpg

fuzz:
fixed_inputs:
replicate_weights: https://replicate.delivery/yhqm/MHJmIF7zlLKXPNN9LKN5yxYUuC7SyzKjMBtqeVwUGdPCpqqJA/trained_model.tar
disabled_inputs:
- extra_lora
duration: 120
iterations: 5

train:
train_timeout: 600
Expand Down
Loading