rvorias's picture
Upload folder using huggingface_hub
b0c7166 verified
config:
name: toastface_prodigy
process:
- datasets:
- cache_latents_to_disk: true
caption_dropout_rate: 0.2
caption_ext: txt
folder_path: /root/lorahub/toastface_prodigy/dataset
resolution:
- 1024
shuffle_tokens: true
token_dropout_rate: 0.01
device: cuda:0
model:
is_flux: true
name_or_path: black-forest-labs/FLUX.1-dev
quantize: true
text_encoder_bits: 8
network:
linear: 42
linear_alpha: 42
transformer_only: true
type: lora
performance_log_every: 500
sample:
guidance_scale: 3.5
height: 1024
neg: ''
prompts:
- Batman [toa$t]
- 1967 Datsun 240z action scene [toa$t]
- a sad rusty robot in the rain [toa$t]
- Mario [toa$t]
- The fist of death [toa$t]
- Anthropomorphic praying mantis [toa$t]
sample_every: 500
sample_steps: 25
sampler: flowmatch
width: 1024
save:
dtype: float16
max_step_saves_to_keep: 3
save_every: 500
save_format: diffusers
train:
batch_size: 1
dtype: bf16
ema_config:
ema_decay: 0.99
use_ema: true
gradient_accumulation_steps: 1
gradient_checkpointing: true
linear_timesteps: true
loss_type: mse
lr: 0.5
noise_scheduler: flowmatch
optimizer: prodigy
reg_weight: 0.5
steps: 5000
target_noise_multiplier: 1.0
train_text_encoder: false
train_unet: true
training_folder: /root/lorahub
trigger_word: C0micz1
type: sd_trainer
job: extension