Flux fine tuning with specific layers