是否支持 flux schnell 的Lora训练

[sp1cae]

因为我看到 AI-Toolkit。配置中有一个标志，可以将其切换为训练 Schnell 而不是 Dev。 如果支持，请问该如何设置? 以下是我的训练参数设置 model_train_type = "flux-lora" pretrained_model_name_or_path = "/lora-scripts/sd-models/flux1-schnell.safetensors" ae = "/lora-scripts/sd-models/ae.safetensors" clip_l = "/lora-scripts/sd-models/clip_l.safetensors" t5xxl = "/lora-scripts/sd-models/t5xxl_fp16.safetensors" timestep_sampling = "sigmoid" sigmoid_scale = 1 model_prediction_type = "raw" discrete_flow_shift = 1 loss_type = "l2" guidance_scale = 1 train_data_dir = "/lora-scripts/train/IMXL_1536X1024" prior_loss_weight = 0 resolution = "1536,1024" enable_bucket = false min_bucket_reso = 256 max_bucket_reso = 2048 bucket_reso_steps = 64 bucket_no_upscale = false output_name = "IM_SCHNELL_V1" output_dir = "./output" save_model_as = "safetensors" save_precision = "bf16" save_every_n_epochs = 1 max_train_epochs = 15 train_batch_size = 2 gradient_checkpointing = true gradient_accumulation_steps = 1 network_train_unet_only = false network_train_text_encoder_only = false learning_rate = 0.0001 unet_lr = 0.0001 text_encoder_lr = 0.00001 lr_scheduler = "cosine_with_restarts" lr_warmup_steps = 0 lr_scheduler_num_cycles = 1 optimizer_type = "AdamW8bit" network_module = "networks.lora_flux" network_dim = 64 network_alpha = 64