File size: 456 Bytes
c8c00f0 | 1 2 3 4 5 6 7 8 9 10 11 12 | import os
from safetensors.torch import load_file
ckpt_path = r"C:\path\to\your\flux1-dev.safetensors" # or wherever your checkpoint is
print(f"File size: {os.path.getsize(ckpt_path) / 1e9:.2f} GB")
state_dict = load_file(ckpt_path)
print(f"Total keys in checkpoint: {len(state_dict)}")
# Check which double_blocks are present
blocks = sorted(set([k.split('.')[1] for k in state_dict.keys() if 'double_blocks' in k]))
print(f"Blocks present: {blocks}") |