Views
No views yet
stabilityai/stable-diffusion-3.5-mediumlr=3e-5, batch_size=16, grad_accum=1, guidance_scale=3.5MeanFlowSD3Transformer2DModel — extends SD3Transformer2DModel with r_embedder submodule for two-time velocity v(x_t, t, r)1import torch
2
3ckpt = torch.load("0010000.pth", map_location="cpu")
4# Navigate checkpoint structure:
5# ckpt['model']['ema_1'] contains the transformer weights
6# Keys are prefixed with 'transformer.' — strip for loading into MeanFlowSD3Transformer2DModel