Running 1 job { "type": "diffusion_trainer", "training_folder": "F:\\ai-toolkit\\AI-Toolkit\\output", "sqlite_db_path": "F:\\ai-toolkit\\AI-Toolkit\\aitk_db.db", "device": "cuda", "trigger_word": "SUP3RM8R1064", "performance_log_every": 10, "network": { "type": "lora", "linear": 64, "linear_alpha": 64, "conv": 16, "conv_alpha": 16, "lokr_full_rank": true, "lokr_factor": -1, "network_kwargs": { "ignore_if_contains": [] } }, "save": { "dtype": "bf16", "save_every": 500, "max_step_saves_to_keep": 2, "save_format": "diffusers", "push_to_hub": false }, "datasets": [ { "folder_path": "F:\\ai-toolkit\\AI-Toolkit\\datasets/supermario64", "mask_path": null, "mask_min_value": 0.1, "default_caption": "", "caption_ext": "txt", "caption_dropout_rate": 0.05, "cache_latents_to_disk": false, "is_reg": false, "network_weight": 1, "resolution": [ 512 ], "controls": [], "shrink_video_to_frames": true, "num_frames": 1, "flip_x": false, "flip_y": false, "num_repeats": 1, "control_path_1": null, "control_path_2": null, "control_path_3": null } ], "train": { "batch_size": 2, "bypass_guidance_embedding": false, "steps": 9000, "gradient_accumulation": 1, "train_unet": true, "train_text_encoder": false, "gradient_checkpointing": true, "noise_scheduler": "flowmatch", "optimizer": "adamw8bit", "timestep_type": "shift", "content_or_style": "balanced", "optimizer_params": { "weight_decay": 0.0001 }, "unload_text_encoder": false, "cache_text_embeddings": false, "lr": 8e-05, "ema_config": { "use_ema": false, "ema_decay": 0.99 }, "skip_first_sample": true, "force_first_sample": false, "disable_sampling": false, "dtype": "bf16", "diff_output_preservation": false, "diff_output_preservation_multiplier": 1, "diff_output_preservation_class": "person", "switch_boundary_every": 1, "loss_type": "mse" }, "logging": { "log_every": 1, "use_ui_logger": true }, "model": { "name_or_path": "F:\\ComfyUI_windows_portable\\ComfyUI\\models\\diffusion_models\\flux-2-klein-base-9b.safetensors", "quantize": true, "qtype": "qfloat8", "quantize_te": true, "qtype_te": "qfloat8", "arch": "flux2_klein_9b", "low_vram": true, "model_kwargs": { "match_target_res": false }, "layer_offloading": false, "layer_offloading_text_encoder_percent": 1, "layer_offloading_transformer_percent": 1 }, "sample": { "sampler": "flowmatch", "sample_every": 500, "width": 1024, "height": 1024, "samples": [ { "prompt": "A SUP3RM8R1064 style 3D platformer game screenshot. Letterboxing. Low quality shading. Shadows are simple circles on the ground. A Mario head, a coin, and a star icon make up the top of the HUD. There is a Camera icon at the bottom right corner of the screenshot. The screenshot depicts a red-capped red-shirted brown-shoed character with a red hat and brown shoes mid-jump and fucking terrified as he makes his way through a crowded Hong Kong city street with traffic and pedestrians as he is chased by a horny stone-gray muscular naked male gargoyle." } ], "neg": "", "seed": 42, "walk_seed": false, "guidance_scale": 4, "sample_steps": 25, "num_frames": 1, "fps": 1 } } Using SQLite database at F:\ai-toolkit\AI-Toolkit\aitk_db.db Job ID: "180aeb72-6da2-48ce-9f21-e07a30e7d811" ############################################# # Running job: Super Mario 64 ############################################# Running 1 process Loading Flux2 model Loading transformer Quantizing Transformer - quantizing 32 transformer blocks 0%| | 0/32 [00:00