ai-toolkit 0.3.6 starts ok - New job start - Lora creation - Krea-2 (raw) - but it is VERY slow. Sometimes it works. No changes.

This is for bugs only

Did you already ask in the discord?

No

You verified that this is a bug and not a feature request or question by asking in the discord?

No

Describe the bug

Geforce 3060 12GB VRAM - 64GB RAM.

Sometimes the job starts and the step took reasonable amout of time... other (same options) look like not using any GPU... same options and don't know how sometime sit works and some times not.

Config File:

job: "extension" config: name: "lea-KREA2" process: - type: "diffusion_trainer" training_folder: "D:\AI\ai-toolkit\AI-Toolkit\output" sqlite_db_path: "./aitk_db.db" device: "cuda" trigger_word: null performance_log_every: 10 network: type: "lora" linear: 32 linear_alpha: 32 lokr_full_rank: true lokr_factor: -1 network_kwargs: ignore_if_contains: [] save: dtype: "bf16" save_every: 250 max_step_saves_to_keep: 4 save_format: "diffusers" push_to_hub: false datasets: - folder_path: "D:\AI\ai-toolkit\AI-Toolkit\datasets/lea" 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 - 768 - 1024 controls: [] shrink_video_to_frames: true num_frames: 1 flip_x: false flip_y: false num_repeats: 1 train: batch_size: 1 bypass_guidance_embedding: false steps: 1500 gradient_accumulation: 1 train_unet: true train_text_encoder: false gradient_checkpointing: true noise_scheduler: "flowmatch" optimizer: "adamw8bit" timestep_type: "linear" content_or_style: "balanced" optimizer_params: weight_decay: 0.0001 unload_text_encoder: false cache_text_embeddings: true lr: 0.0001 ema_config: use_ema: false ema_decay: 0.99 skip_first_sample: false force_first_sample: false disable_sampling: true 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: "krea/Krea-2-Raw" quantize: true qtype: "qfloat8" quantize_te: true qtype_te: "qfloat8" arch: "krea2" low_vram: true model_kwargs: {} compile: false layer_offloading: false layer_offloading_text_encoder_percent: 1 layer_offloading_transformer_percent: 1 sample: sampler: "flowmatch" sample_every: 250 sample_start_step: 0 width: 1024 height: 1024 samples: - prompt: "woman with red hair, playing chess at the park, bomb going off in the background" - prompt: "a woman holding a coffee cup, in a beanie, sitting at a cafe" - prompt: "a horse is a DJ at a night club, fish eye lens, smoke machine, lazer lights, holding a martini" - prompt: "a man showing off his cool new t shirt at the beach, a shark is jumping out of the water in the background" - prompt: "a bear building a log cabin in the snow covered mountains" - prompt: "woman playing the guitar, on stage, singing a song, laser lights, punk rocker" - prompt: "hipster man with a beard, building a chair, in a wood shop" - prompt: "photo of a man, white background, medium shot, modeling clothing, studio lighting, white backdrop" - prompt: "a man holding a sign that says, 'this is a sign'" - prompt: "a bulldog, in a post apocalyptic world, with a shotgun, in a leather jacket, in a desert, with a motorcycle" neg: "" seed: 42 walk_seed: true guidance_scale: 4 sample_steps: 30 num_frames: 1 fps: 1 meta: name: "[name]" version: "1.0"

Any help will be more than welcome! Regards, Javy.