{
  "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"
  ]
}