Dan Saunders 1b53c49e1a text diffusion training plugin (#3067) ...

* diffusion training plugin

* cleanup

* nits

* fixes + improvements

* add back in reinit_weights (clobbered?); masking / pretrain fixes

* nits

* cleanup; tests draft

* sample generation, tests fixes

* fixes

* nits

* add inference support; add auto-mask token support

* nits

* nits

* progress

* simplify logging

* lint

* prefix args with diffusion_

* coderabbito

* tests fix

* nit

* nits

* cleanup + nits

* nits

* fix SFT sample gen

* fixes

* fix

* comments

* comments

* lint

* reward model lora fix

* cleanup; fix pretraining_dataset case

* gradio inference

* update cfgs

* update cfgs

* train, generation parity, cleanup

* fix

* simplify

* test

* test fix

2025-09-10 20:27:00 -04:00

..

alst

Distributed/ND-Parallel (#2977)

2025-07-31 15:25:02 -04:00

arcee

Feat: add arcee (#3028)

2025-08-08 08:09:11 -04:00

archived

drop old patches and code that are no longer needed (#3007) [skip ci]

2025-08-06 08:02:39 -04:00

cloud

deploy training jobs to baseten w truss in axolotl cli (#3086) [skip ci]

2025-08-26 09:29:50 -04:00

cohere

make pad_to_sequence_len default to the same value as sample_packing (#2941) [skip ci]

2025-07-21 11:40:56 -04:00

colab-notebooks

text diffusion training plugin (#3067)

2025-09-10 20:27:00 -04:00

deepcogito

use warmup_ratio as a better default than warmup steps since it's data dependent (#2897) [skip ci]

2025-07-30 06:44:06 -04:00

deepseek-v2

use warmup_ratio as a better default than warmup steps since it's data dependent (#2897) [skip ci]

2025-07-30 06:44:06 -04:00

devstral

Feat: add hunyuan v1 (#3016)

2025-09-10 09:03:30 +07:00

distributed-parallel

feat: update nd parallelism readme (#3039)

2025-08-08 12:45:36 +01:00

falcon-h1

make pad_to_sequence_len default to the same value as sample_packing (#2941) [skip ci]

2025-07-21 11:40:56 -04:00

gemma2

make pad_to_sequence_len default to the same value as sample_packing (#2941) [skip ci]

2025-07-21 11:40:56 -04:00

gemma3

Create 270m-qlora.yml (#3075) [skip ci]

2025-09-03 16:20:32 -04:00

gemma3n

feat(doc): standardize the axolotl install to a release (#3040) [skip ci]

2025-08-08 08:00:26 -04:00

glm4

use warmup_ratio as a better default than warmup steps since it's data dependent (#2897) [skip ci]

2025-07-30 06:44:06 -04:00

gpt-oss

upgrade flash-attn to 2.8.3 for gpt-oss attn sink support (#3082)

2025-08-21 15:04:10 -04:00

hunyuan

Feat: add hunyuan v1 (#3016)

2025-09-10 09:03:30 +07:00

jamba

use warmup_ratio as a better default than warmup steps since it's data dependent (#2897) [skip ci]

2025-07-30 06:44:06 -04:00

LiquidAI

Various fixes for VLMs (#3063)

2025-08-15 10:52:57 -04:00

llama-2

drop old patches and code that are no longer needed (#3007) [skip ci]

2025-08-06 08:02:39 -04:00

llama-3

text diffusion training plugin (#3067)

2025-09-10 20:27:00 -04:00

llama-3-vision

fix: revert changing default optimizer to muon (#2965) [skip ci]

2025-07-22 10:00:30 -04:00

llama-4

use warmup_ratio as a better default than warmup steps since it's data dependent (#2897) [skip ci]

2025-07-30 06:44:06 -04:00

llava

fix: revert changing default optimizer to muon (#2965) [skip ci]

2025-07-22 10:00:30 -04:00

magistral

Feat: add hunyuan v1 (#3016)

2025-09-10 09:03:30 +07:00

mamba

use warmup_ratio as a better default than warmup steps since it's data dependent (#2897) [skip ci]

2025-07-30 06:44:06 -04:00

mistral

use warmup_ratio as a better default than warmup steps since it's data dependent (#2897) [skip ci]

2025-07-30 06:44:06 -04:00

orpheus

use warmup_ratio as a better default than warmup steps since it's data dependent (#2897) [skip ci]

2025-07-30 06:44:06 -04:00

phi

use warmup_ratio as a better default than warmup steps since it's data dependent (#2897) [skip ci]

2025-07-30 06:44:06 -04:00

pixtral

fix: revert changing default optimizer to muon (#2965) [skip ci]

2025-07-22 10:00:30 -04:00

qwen2

use warmup_ratio as a better default than warmup steps since it's data dependent (#2897) [skip ci]

2025-07-30 06:44:06 -04:00

qwen2_5-vl

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

qwen2-vl

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

qwen3

Center rewards coefficient (#3124)

2025-09-03 16:22:37 -04:00

seed-oss

Feat: add seedoss (#3104) [skip ci]

2025-09-10 09:01:02 +07:00

slurm

Example for Slurm and various fixes (#3038) [skip ci]

2025-08-08 08:02:03 -04:00

smolvlm2

Various fixes for VLMs (#3063)

2025-08-15 10:52:57 -04:00

streaming

Streaming SFT support (#3101)

2025-09-02 12:08:44 -04:00

voxtral

Feat: add hunyuan v1 (#3016)

2025-09-10 09:03:30 +07:00