[transformer v2] Load + evaluate FID for OpenAI guided-diffusion models - #84

Open

Birch-san wants to merge 7 commits into

crowsonkb:transformer-model-v2from

Commits on Oct 17, 2023

support loading OpenAI guided-diffusion models for evaluation. reduce overhead of evaluate_only mode (skip construction/accelerator-preparation of trainable model, optimizer, schedules)
Birch-san
committed
51a5310
prefer inference_mode context for demo and eval, due to lower overhead
Birch-san
committed
5ee74be
sample density for OpenAI models
Birch-san
committed
bd8372e
move OpenAIVDenoiser into k-diffusion library
Birch-san
committed
61f52c6
change class-cond key to match underlying model's expectations
Birch-san
committed
ddfa060
delete launch config not relevant to PR
Birch-san
committed
a57b6fc

Commits on Oct 18, 2023

support torch sdp attn for OpenAI guided diffusion
Birch-san
committed
bfd13d8