Views
No views yet
| file | role |
|---|---|
unet/model_io32.onnx | graph — fp32 I/O, INT64 timestep, fp16 weights/compute (the profile validated on WebGPU) |
unet/model.onnx | graph — original fp16-I/O self-export profile (kept for reference) |
unet/model.onnx_data, _1, _2 | weight shards (<2 GB each; shared by both graphs) |
sample float32 [B,4,H,W] · timestep int64 [steps] · encoder_hidden_states
float32 [B,77,2048] · text_embeds float32 [B,1280] · time_ids float32 [B,6] →
out_sample float32 [B,4,H,W].LIGHTNING_STEPS=0),
retargeted with tools/retarget-unet-profile.py; one UNet pass agrees with the
torch fp16 checkpoint at cosine 1.000000.