Dependencies versions: - transformers: 4.49.0 - datasets: 3.3.2 - accelerate: 1.4.0 - torch: 2.5.1 - diffusers: 0.32.2 - trl: 0.15.1 - peft: 0.14.0 - [ x] Training - GPU - [ x] Inference - CPU - [x ] Inference - GPU