PEFT-method-comparison-embed / image-gen /default_training_params.json
github-actions[bot]
🚀 Deploy method comparison app from GH action
2b239c3
Raw
History Blame Contribute Delete
2.67 kB
{
"model_id": "black-forest-labs/FLUX.2-klein-base-4B",
"dataset_id": "peft-internal-testing/cat-image-dataset",
"dataset_split": "train",
"dtype": "bfloat16",
"resolution": 512,
"batch_size": 2,
"batch_size_eval": 1,
"repeats": 1000,
"max_steps": 750,
"eval_steps": 100,
"compile": false,
"use_gc": true,
"seed": 0,
"grad_norm_clip": 1.0,
"optimizer_type": "AdamW",
"optimizer_kwargs": {
"lr": 0.0001,
"weight_decay": 0.0001
},
"lr_scheduler": null,
"use_amp": false,
"autocast_adapter_dtype": true,
"image_column": "image",
"valid_size": 2,
"test_size": 4,
"num_inference_steps": 20,
"guidance_scale": 3.5,
"max_sequence_length": 512,
"text_encoder_out_layers": [10, 20, 30],
"weighting_scheme": "none",
"logit_mean": 0.0,
"logit_std": 1.0,
"mode_scale": 1.29,
"dino_model_id": "facebook/dinov2-base",
"dino_image_size": 224,
"instance_prompts": [
"sks cat sitting on a chair in front of a box of chocolates",
"sks cat playing on the Steam Deck",
"sks cat wearing a necklace while sitting in a box on a sofa",
"a box with four donuts in front of sks cat",
"sks cat wearing a pink veil with flowers on it and a dagger made out of yellow cardboard",
"sks cat between two pillows, with one pillow showing a polar bear and the other a fox",
"sks cat with an espresso reading the newspaper",
"a close up of a hand petting sks cat on the head",
"sks cat opens a white door",
"sks cat with a VIP \"all access\" pass",
"sks cat in front of a monitor showing the video game \"RimWorld\"",
"sks cat reads a newspaper about cannabis",
"sks cat reading the newspaper with the title \"DIE BOX\"",
"sks cat being fed sushi",
"sks cat in front of a box of tomatoes",
"a gingerbread house in front of sks cat",
"sks cat is wearing a Wednesday Addams costume",
"sks cat looks skeptical as he reads a magazine depicting a black cat",
"sks cat wearing a striped scarf",
"sleepy sks cat lying in bed covered by a blue blanket"
],
"sample_image_prompts": [
"a photo of sks cat",
"a color drawing of sks cat",
"sks cat at the beach",
"a photo of sks dog",
"a photo of an orange cat"
],
"drift_image_prompts": [
"a photo of cat",
"a color drawing of cat",
"cat sitting in a bathtub",
"cat looks into a mirror",
"a photo of cat next to a tree",
"the Mona Lisa but she looks angry",
"a dog holding a sign that says hello world",
"a photo of the Eiffel tower at night",
"an astronaut riding a horse",
"3d design concept of a chair that looks like an avocado"
]
}